Pagini recente » Cod sursa (job #1650431) | Cod sursa (job #1650328) | Statistici casir viorel (viorelus) | Cod sursa (job #984311) | Cod sursa (job #2208631)
#include <iostream>
#include <fstream>
#include <vector>
#define NMAX 100005
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
int n,i,j,a,b,nre,q[NMAX],k[NMAX],rz[NMAX];
vector<int> vc[NMAX];
void dfs(int n)
{
q[nre++]=n;
if(k[n]!=0) rz[n]=rz[q[nre-k[n]-1]]+1;
for(int i=0;i<vc[n].size();i++)
dfs(vc[n][i]);
nre--;
}
int main() {
fin>>n;
for(i=1;i<=n;i++)
fin>>k[i];
for(i=1;i<n;i++)
{
fin>>a>>b;
vc[a].push_back(b);
rz[b]=1;
}
for(i=1;i<=n;i++)
{
if(rz[i]==0) a=i;
rz[i]=0;
}
nre=1;
dfs(a);
for(i=1;i<=n;i++)
fout<<rz[i]<<" "; fout<<"\n";
}