Cod sursa(job #926991)

Utilizator raulstoinStoin Raul raulstoin Data 25 martie 2013 15:15:35
Problema Cerere Scor 50
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.64 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;
			int k=0;
			while(v[x])
			{
				for(int i=x,aux=v[i];aux;aux--)
					x=TT[x];
				k++;
			}
			A[i]=k;
		}
}
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;
}