Cod sursa(job #2208631)

Utilizator shantih1Alex S Hill shantih1 Data 30 mai 2018 19:23:46
Problema Cerere Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.59 kb
#include <iostream>
#include <fstream>
#include <vector>
#define NMAX 100005

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

int n,i,j,a,b,nre,q[NMAX],k[NMAX],rz[NMAX];
vector<int> vc[NMAX];

void dfs(int n)
{
	q[nre++]=n;
	if(k[n]!=0)	rz[n]=rz[q[nre-k[n]-1]]+1;
	
	for(int i=0;i<vc[n].size();i++)
		dfs(vc[n][i]);
	nre--;
}
int main() {
	
	fin>>n;
	for(i=1;i<=n;i++)
		fin>>k[i];
	for(i=1;i<n;i++)
	{
		fin>>a>>b;
		vc[a].push_back(b);
		rz[b]=1;
	}
	for(i=1;i<=n;i++)
	{
		if(rz[i]==0)	a=i;
		rz[i]=0;
	}
	nre=1;
	dfs(a);
	for(i=1;i<=n;i++)
		fout<<rz[i]<<" ";	fout<<"\n";
}