Cod sursa(job #2208629)

Utilizator shantih1Alex S Hill shantih1 Data 30 mai 2018 19:13:23
Problema Cerere Scor 85
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,k[NMAX],rz[NMAX];
vector<int> vc[NMAX],q;

void dfs(int n)
{
	q.push_back(n);
	if(k[n]==0)	rz[n]=0;
	else rz[n]=rz[q[q.size()-k[n]-1]]+1;
	
	for(int i=0;i<vc[n].size();i++)
		dfs(vc[n][i]);
	q.pop_back();
}
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;
	
	dfs(a);
	for(i=1;i<=n;i++)
		fout<<rz[i]<<" ";	fout<<"\n";
}