Pagini recente » Cod sursa (job #2946653) | Cod sursa (job #2171919) | Cod sursa (job #318960) | Cod sursa (job #2637696) | Cod sursa (job #1897937)
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
vector <int> G[100005];
int N,dp[100005],st[100005],nst,K[100005],sursa;
void Dfs(int nod)
{
if(!K[nod])
dp[nod]=0;
else
dp[nod]=dp[st[nst-K[nod]+1]]+1;
st[++nst]=nod;
for(int i=0;i<G[nod].size();i++)
{
int son=G[nod][i];
Dfs(son);
}
nst--;
}
int main()
{
fin>>N;
for(int i=1;i<=N;i++)
{
fin>>K[i];
}
for(int i=1;i<N;i++)
{
int x,y;
fin>>x>>y;
G[x].push_back(y);
dp[y]=1;
}
for(int i=1;i<=N;i++)
{
if(!dp[i])
sursa=i;
dp[i]=0;
}
Dfs(sursa);
for(int i=1;i<=N;i++)
{
fout<<dp[i]<<" ";
}
return 0;
}