Cod sursa(job #391904)

Utilizator bog29Antohi Bogdan bog29 Data 6 februarie 2010 14:20:22
Problema Cerere Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.65 kb
#include<fstream>
#include<vector>
#define dmax 100003
using namespace std;
ifstream in("cerere.in");
ofstream out("cerere.out");

int n,c[dmax],t[dmax],dm[dmax],stv[dmax];
vector<int>g[dmax];

void dfs(int k,int pas)
{	vector<int>::iterator it;
	stv[pas]=k;
	if(c[k]!=0)
		dm[k]=1+dm[stv[pas-c[k]]];
	for(it=g[k].begin();it<g[k].end();it++)
		dfs(*it,pas+1);	
}


int main()
{	int i,a,b,st;
	in>>n;
	for(i=1;i<=n;i++)
		in>>c[i];
	for(i=1;i<n;i++)
	{	in>>a>>b;	
		t[b]=a;
		g[a].push_back(b);
	}
	in.close();
	for(i=1;i<=n;i++)
		if(t[i]==0)st=i;
	dfs(st,1);
	for(i=1;i<=n;i++)
		out<<dm[i]<<" "; 
	out.close();
	return 0;
}