Cod sursa(job #807560)

Utilizator ichigo2908mantu radu ichigo2908 Data 4 noiembrie 2012 22:20:19
Problema Cerere Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.65 kb
#include <fstream>
#include <vector>
using namespace std;

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

const int N=100001;
vector <int> v[N];
int n, k[N], sol[N], a[N];

void citire()
{
	int iTemp1, iTemp2;
	
	in>>n;
	for(int i=1;i<=n;i++)
		in>>k[i];
	for(int i=0;i<n;i++)
	{
		in>>iTemp1>>iTemp2;
		v[iTemp1].push_back(iTemp2);
	}
}

void dfs(size_t x, int lvl)
{
	if(k[x]==0)
		{
			sol[x]=0;
			a[lvl]=0;
		}
		else
		{
			sol[x]=a[lvl-k[x]]+1;
			a[lvl]=a[lvl-k[x]]+1;
		}
	for(size_t i=0;i<v[x].size();i++)
		dfs(v[x][i],lvl+1);
}

int main()
{
	citire();
	dfs(1,0);

	for(int i=1;i<=n;i++)
		out<<sol[i]<<' ';

	in.close();
	out.close();
	
	return 0;
}