Cod sursa(job #32644)

Utilizator razvi9Jurca Razvan razvi9 Data 18 martie 2007 11:36:46
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,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;}