Pagini recente » Cod sursa (job #1623121) | Cod sursa (job #2690967) | Cod sursa (job #1069259) | Cod sursa (job #2361134) | Cod sursa (job #32646)
Cod sursa(job #32646)
#include<stdio.h>
int t[100001],a[100001],n,i,x,y,k[100001],j,nr;
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++)
if(k[i])
{y=i;x=k[i];nr=0;
while(x){x--; y=t[y];if(!x){nr++; if(a[y]) {nr=nr+a[y]-1;break;}x=k[y];}}
a[i]=nr+1;
printf("%d ",nr);}
else {a[i]=1;printf("0 ");}
fclose(stdout);
return 0;}