Cod sursa(job #1069602)

Utilizator federerUAIC-Padurariu-Cristian federer Data 30 decembrie 2013 12:24:14
Problema Cerere Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.73 kb
#include<fstream>
#include<vector>
#define Nmax 100001
using namespace std;

ifstream fin("cerere.in");
ofstream fout("cerere.out");

vector <int> V[Nmax];
int sol[Nmax], K[Nmax], root;
int st[Nmax];

void dfs(int nod, int k)
{
	st[k] = nod;
	if (K[nod])
		sol[nod] = sol[st[k - K[nod]]] + 1;

	for (int i = 0; i < V[nod].size(); ++i)
		dfs(V[nod][i], k + 1);
}

int main()
{
	int x, y, N, i;
	fin >> N;
	for (i = 1; i <= N; ++i)
	{
		fin >> K[i];
		if (K[i])
			sol[i] = 1;
	}
	root = N;
	for (i = 1; i < N; ++i)
	{
		fin >> x >> y;
		V[x].push_back(y);
		root ^= y ^ i;
	}
	dfs(root, 0);
	for (i = 1; i <= N; ++i)
		fout << sol[i] << ' ';
	fout << '\n';
	fin.close();
	fout.close();
	return 0;
}