Cod sursa(job #1220004)

Utilizator yololy97Olaru Bogdan-Ioan yololy97 Data 16 august 2014 12:11:52
Problema Cerere Scor 65
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.84 kb
#include <cstdio>
#include <vector>
#define N 100001
using namespace std;
int n, k[N], d[N];
vector<int> a[N];
vector<int> b;
void read(){
	scanf("%d ", &n);
	for(int i = 1; i <= n; ++i)
		scanf("%d ", &k[i]);
	int x, y;
	for(int i = 1; i < n; ++i){
		scanf("%d %d ", &x, &y);
		//a[y].push_back(x);
		a[x].push_back(y);
	}
}
bool used[N];
void dfs(int t){
	used[t] = true;
	b.push_back(t);
	d[t] = d[b[b.size() - k[t] - 1]] + 1;
	for(vector<int>::iterator it = a[t].begin(); it != a[t].end(); ++it)
		if(!used[*it])
			dfs(*it);
	b.pop_back();
}
void solve(){
	for(int i = 1; i <= n; ++i)
		if(!k[i]){
			dfs(i);
			break;
		}
	for(int i = 1; i <= n; ++i)
		printf("%d ", d[i] - 1);
	printf("\n");
}
int main(){
	freopen("cerere.in", "r", stdin);
freopen("cerere.out", "w", stdout);
	read();
	solve();
	return 0;
}