Pagini recente » Cod sursa (job #1601982) | Cod sursa (job #214261) | Cod sursa (job #2960609) | Cod sursa (job #946134) | Cod sursa (job #2283775)
#include <bits/stdc++.h>
using namespace std;
ifstream in("cerere.in");
ofstream out("cerere.out");
vector< int > v, s, ans;
vector< bool > viz;
vector< vector < int > > G;
void DFS(int node) {
viz[node] = true;
s.push_back(node);
for(auto vecin: G[node]) {
if(!viz[vecin]) {
if(v[vecin] != 0) {
ans[vecin] = ans[s[(int)s.size() - v[vecin]]] + 1;
}
DFS(vecin);
}
}
s.pop_back();
}
int main() {
ios::sync_with_stdio(false); in.tie(0); out.tie(0);
int n; in >> n;
v.resize(n + 1); ans.resize(n + 1); viz.resize(n + 1, false); G.resize(n + 1, vector< int >());
for(int i = 1; i <= n; ++i) {
in >> v[i];
}
vector< bool > isRoot(n + 1, true);
for(int i = 1; i < n; ++i) {
int a, b; in >> a >> b;
G[a].push_back(b);
isRoot[b] = false;
}
for(int i = 1; i <= n; ++i) {
if(isRoot[i]) {
DFS(i);
}
}
for(int i = 1; i <= n; ++i) {
out << ans[i] << " ";
}
out << "\n";
in.close(); out.close();
return 0;
}