Pagini recente » Cod sursa (job #1543763) | Cod sursa (job #2786532) | Cod sursa (job #1055292) | Cod sursa (job #2536215) | Cod sursa (job #2283243)
#include <bits/stdc++.h>
using namespace std;
ifstream in("cerere.in");
ofstream out("cerere.out");
const int inf = 1e9;
vector< int > v, ans;
vector< bool > viz;
vector< vector < int > > G;
void DFS() {
stack< int > s; s.push(1);
while(!s.empty()) {
int node = s.top();
viz[node] = true;
bool toEliminate = true;
for(auto vecin: G[node]) {
if(!viz[vecin]) {
toEliminate = false;
if(v[vecin]) {
stack< int > temp;
for(int k = 1; k <= v[vecin]; ++k) {
temp.push(s.top()); s.pop();
}
ans[vecin] = ans[temp.top()] + 1;
while(!temp.empty()) {
s.push(temp.top()); temp.pop();
}
}
s.push(vecin);
goto eliminate;
}
}
eliminate:
if(toEliminate) {
s.pop();
}
}
}
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, inf); viz.resize(n + 1, false); G.resize(n + 1, vector< int >());
for(int i = 1; i <= n; ++i) {
in >> v[i];
if(v[i] == 0) {
ans[i] = 0;
}
}
for(int i = 1; i < n; ++i) {
int a, b; in >> a >> b;
G[a].push_back(b);
}
DFS();
for(int i = 1; i <= n; ++i) {
out << ans[i] << " ";
}
out << "\n";
in.close(); out.close();
return 0;
}