Cod sursa(job #39372)

Utilizator razvi9Jurca Razvan razvi9 Data 26 martie 2007 17:53:18
Problema Cerere Scor 50
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.53 kb
#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[1000000];
 setvbuf(stdin,buf,_IOFBF,1000000);
 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;}