Pagini recente » Cod sursa (job #2695017) | Cod sursa (job #1046562) | Cod sursa (job #610782) | Cod sursa (job #1960479) | Cod sursa (job #309287)
Cod sursa(job #309287)
#include <cstdio>
#include <vector>
using namespace std;
#define Nmax 100100
#define file_in "cerere.in"
#define file_out "cerere.out"
#define sz size
#define pb push_back
vector<int> v[Nmax];
int grad[Nmax];
int n,k[Nmax];
int nodp;
int vf[Nmax];
int x,y;
int sol[Nmax];
void dfs(int nod)
{
int i;
vf[vf[0]]=nod;
sol[nod]=sol[vf[vf[0]-k[nod]]]+1;
if (k[nod]==0)
sol[nod]=0;
for (i=0;i<v[nod].sz();++i)
{
vf[0]++;
dfs(v[nod][i]);
vf[0]--;
}
}
void citire()
{
int i;
freopen(file_in,"r",stdin);
freopen(file_out,"w",stdout);
scanf("%d", &n);
for (i=1;i<=n;++i)
scanf("%d", &k[i]);
for (i=1;i<n;++i)
{
scanf("%d %d", &x,&y);
v[x].pb(y);
++grad[y];
}
}
int nod()
{
int i;
for (i=1;i<=n;++i)
if (grad[i]==0)
return i;
}
int main()
{
int i;
citire();
nodp=nod();
vf[0]=1;
dfs(nodp);
for (i=1;i<=n;++i)
printf("%d ",sol[i]);
fclose(stdin);
fclose(stdout);
return 0;
}