Pagini recente » Cod sursa (job #816368) | Cod sursa (job #2119446) | Cod sursa (job #186273) | Cod sursa (job #3164597) | Cod sursa (job #2542082)
#include <bits/stdc++.h>
using namespace std;
//ifstream f("memory001.in");
//ofstream g("memory001.out");
ifstream f("cerere.in");
ofstream g("cerere.out");
const int dim = 100001;
int n;
int k[dim];
vector <int> sons[dim];
int parent[dim];
int dist[dim];
bool mark[dim];
void dfs(int node){
mark[node] = 1;
for(int it: sons[node]){
if(!mark[it]){
if(k[it] > 0){
int p = it;
for(int i = 1; i <= k[it]; ++i)
p = parent[p];
dist[it] = dist[p] + 1;
}
dfs(it);
}
}
}
int main()
{
int i, j, a, b;
f >> n;
for(i = 1; i <= n; ++i){
f >> k[i];
}
for(i = 1; i < n; ++i){
f >> a >> b;
sons[a].push_back(b);
parent[b] = a;
}
int root;
for(i = 1; i <= n; ++i)
if(parent[i] == 0){
root = i;
break;
}
dfs(root);
for(i = 1; i <= n; ++i)
g << dist[i] << " ";
return 0;
}