Cod sursa(job #926521)

Utilizator raulstoinStoin Raul raulstoin Data 25 martie 2013 11:36:53
Problema Cerere Scor 25
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.84 kb
#include<fstream>
#include<algorithm>
#define NMAX 100005
using namespace std;
int n,v[NMAX],str[18][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;
		str[0][y]=x;
	}
	fin.close();
}
inline void stramosi()
{
	for(int i=1;i<=n;i++)
		for(int j=1;(1<<j)<=n;j++)
			str[j][i]=str[j-1][str[j-1][i]];
}
inline int query(int x,int y)
{
	if(!y)
		return x;
	int i;
	for(i=0;(1<<i)<=y;i++);
	i--;
	y-=(1<<i);
	x=str[i][x];
	return query(x,y);
}
void solve()
{
	stramosi();
	for(int i=1,x;i<=n;i++)
		if(v[i])
		{
			x=query(i,v[i]);
			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;
}