Pagini recente » Cod sursa (job #326270) | Cod sursa (job #211608) | Cod sursa (job #1406202) | Cod sursa (job #2706371) | Cod sursa (job #2907007)
/// [A][M][C][B][N] ///
#include <bits/stdc++.h>
using namespace std;
const int mod = 1e9 + 7, inf = 0x3f3f3f3f;
const char sp = ' ', nl = '\n';
ifstream fin("cerere.in");
ofstream fout("cerere.out");
const int nmax = 1e5;
int n, v[nmax + 1], dp[nmax + 1];
vector<int> g[nmax + 1], path;
void dfs(int node) {
if (v[node]) {
dp[node] = dp[path[path.size() - v[node]]] + 1;
}
path.push_back(node);
for (auto& son : g[node]) {
dfs(son);
}
path.pop_back();
}
int main() {
ios::sync_with_stdio(0);
cin.tie(0), cout.tie(0);
fin >> n;
set<int> s;
for (int i = 1; i <= n; ++i) {
fin >> v[i];
s.insert(i);
}
for (int i = 1; i < n; ++i) {
int x, y;
fin >> x >> y;
g[x].push_back(y);
s.erase(y);
}
dfs(*s.begin());
for (int i = 1; i <= n; ++i) {
fout << dp[i] << sp;
}
}