Pagini recente » Atasamentele paginii Profil lazaralex2002 | Cod sursa (job #3365718) | Cod sursa (job #3366873) | Cod sursa (job #3365719) | Cod sursa (job #3365677)
#include <bits/stdc++.h>
using namespace std;
int nodes[100005], ans[100005], pas[100005], par[100005];
int sz = 0;
vector<int> adj[100005];
void dfs(int i){
nodes[++sz] = i;
if(pas[i])
ans[i] = ans[nodes[sz - pas[i]]] + 1;
for(int j : adj[i])
dfs(j);
sz--;
}
int main(){
//ifstream cin("cerere.in");
//ofstream cout("cerere.out");
int n;
cin >> n;
for(int i = 1; i <= n; i++){
cin >> pas[i];
}
int root = 0;
for(int i = 1; i < n; i++){
int u, v;
cin >> u >> v;
adj[u].push_back(v);
par[v] = u;
}
for(int i = 1; i <= n; i++){
if(par[i] == 0) root = i;
}
dfs(root);
for(int i = 1; i <= n; i++) cout << ans[i] << ' ';
return 0;
}