Cod sursa(job #807583)

Utilizator ichigo2908mantu radu ichigo2908 Data 4 noiembrie 2012 22:56:09
Problema Cerere Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.69 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, sol[N], k[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-1;i++)
	{
		in>>iTemp1>>iTemp2;
		v[iTemp1].push_back(iTemp2);
	}
}

void dfs(size_t x, int lvl)
{
	if(!k[x])
	{
		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;
}