Pagini recente » Cod sursa (job #1858293) | Cod sursa (job #1799782) | Cod sursa (job #965024) | Cod sursa (job #2565752) | Cod sursa (job #1326188)
#include<cstdio>
#include<vector>
#define NMAX 100005
using namespace std;
int a[NMAX],sol[NMAX];
int viz[NMAX],str[NMAX],st[NMAX];
int i,x,y,n,k,start;
vector<int>v[NMAX];
void DFS(int nod)
{
viz[nod]=1;
st[++k]=nod;
if (a[nod]!=0) sol[nod]=sol[st[k-a[nod]]]+1;
for (int i=0;i<v[nod].size();++i)
{
int it=v[nod][i];
if (viz[it]==0) DFS(it);
}
--k;
}
int main()
{
freopen("cerere.in","r",stdin);
freopen("cerere.out","w",stdout);
scanf("%d",&n);
for (i=1;i<=n;++i) scanf("%d",&a[i]);
for (i=1;i<=n-1;++i)
{
scanf("%d%d",&x,&y);
v[x].push_back(y);
str[y]=1;
}
for (i=1;i<=n;++i)
if (str[i]==0) start=i;
DFS(start);
for(i=1;i<=n;++i) printf("%d ",sol[i]);
return 0;
}