Cod sursa(job #983304)

Utilizator Kira96Denis Mita Kira96 Data 11 august 2013 14:55:14
Problema Cerere Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.55 kb
#include<fstream>
#define N 100100
#include<vector>
using namespace std;
ifstream f("cerere.in");
ofstream g("cerere.out");
int sol[N],s[N],i,x,y,t,T[N],k[N],n;
vector<int> v[N];
void dfs(int x)
{
	s[++t]=x;
	if(k[x])
	sol[x]=sol[s[t-k[x]]]+1;
	for(int i=0;i<v[x].size();++i)
		dfs(v[x][i]);
	--t;
}
int main ()
{
	f>>n;
	for(i=1;i<=n;++i)
		f>>k[i];
	for(i=1;i<n;++i)
	{
		f>>x>>y;
		v[x].push_back(y);
		T[y]=1;
	}
	for(i=1;i<=n;++i)
		if(!T[i])
		{
			dfs(i);
			break;
		}
	for(i=1;i<=n;++i)
		g<<sol[i]<<" ";
	return 0;
}