Cod sursa(job #309287)

Utilizator AndreiDDiaconeasa Andrei AndreiD Data 29 aprilie 2009 23:18:09
Problema Cerere Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.02 kb
#include <cstdio>
#include <vector>

using namespace std;

#define Nmax 100100

#define file_in "cerere.in"
#define file_out "cerere.out"

#define sz size
#define  pb push_back

vector<int> v[Nmax];
int grad[Nmax];
int n,k[Nmax];
int nodp;
int vf[Nmax];
int x,y;
int sol[Nmax];


void dfs(int nod)
{
	int i;
	vf[vf[0]]=nod;
	sol[nod]=sol[vf[vf[0]-k[nod]]]+1;
	if (k[nod]==0)
		sol[nod]=0; 
	for (i=0;i<v[nod].sz();++i)
	{
		vf[0]++;
		dfs(v[nod][i]);
		vf[0]--;
	}
}
		

void citire()
{
	int i;
	
	freopen(file_in,"r",stdin);
	freopen(file_out,"w",stdout);
	
	scanf("%d", &n);
	for (i=1;i<=n;++i)
		 scanf("%d", &k[i]);
	
	for (i=1;i<n;++i)
	{
		scanf("%d %d", &x,&y);
		v[x].pb(y);
		++grad[y];
	}
}

int nod()
{
	int i;
	for (i=1;i<=n;++i)
		 if (grad[i]==0)
		     return i;
}		 
           
int main()
{
	int i;
	citire();
	nodp=nod();
	vf[0]=1;
	dfs(nodp);
	
	for (i=1;i<=n;++i)
		printf("%d ",sol[i]);
		
	fclose(stdin);
    fclose(stdout);

	return 0;
}