Pagini recente » Cod sursa (job #832758) | Cod sursa (job #1732355) | Cod sursa (job #1110807) | Cod sursa (job #1040796) | Cod sursa (job #32644)
Cod sursa(job #32644)
#include<stdio.h>
int t[100001],a[100001],n,i,x,y,k[100001],j,nr,t2[100001];
int main()
{freopen("cerere.in","r",stdin);
freopen("cerere.out","w",stdout);
scanf("%d",&n);
for(i=1;i<=n;i++)
scanf("%d",&k[i]);
for(j=1;j<n;j++)
{scanf("%d %d",&y,&x);t[x]=y;}
for(i=1;i<=n;i++)
{x=k[i];t2[i]=i;
while(x){t2[i]=t[t2[i]];x--;}}
for(i=1;i<=n;i++)
if(k[i])
{x=i;nr=0;
while(k[x]){nr++;x=t2[x];if(a[x]){nr=nr+a[x];break;}}
a[i]=nr;
printf("%d ",nr);
}
else printf("0 ");
fclose(stdout);
return 0;}