Cod sursa(job #926974)

Utilizator raulstoinStoin Raul raulstoin Data 25 martie 2013 15:01:17
Problema Cerere Scor 5
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.57 kb
#include<fstream>
#include<algorithm>
#define NMAX 100005
using namespace std;
int n,v[NMAX],TT[NMAX],A[NMAX];
void read()
{
	ifstream fin("cerere.in");
	fin>>n;
	int i,x,y;
	for(i=1;i<=n;i++)
		fin>>v[i];
	for(i=1;i<n;i++)
	{
		fin>>x>>y;
		TT[y]=x;
	}
	fin.close();
}
void solve()
{
	for(int i=1,x;i<=n;i++)
		if(v[i])
		{
			x=i;
			while(v[i]--)
				x=TT[x];
			A[i]=A[x]+1;
		}
}
void print()
{
	ofstream fout("cerere.out");
	for(int i=1;i<=n;i++)
		fout<<A[i]<<' ';
	fout.close();
}
int main()
{
	read();
	solve();
	print();
	return 0;
}