Pagini recente » Cod sursa (job #288829) | Cod sursa (job #2076761) | Cod sursa (job #2427038) | Cod sursa (job #140547) | Cod sursa (job #39374)
Cod sursa(job #39374)
#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);
char buf[500000];
setvbuf(stdin,buf,_IOFBF,500000);
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];break;}x=k[y];}}
a[i]=nr;
printf("%d ",nr);}
else {printf("0 ");}
fclose(stdout);
return 0;}