Pagini recente » Cod sursa (job #341093) | Cod sursa (job #3240662) | Cod sursa (job #1951126) | Cod sursa (job #302160) | Cod sursa (job #2494194)
#include <bits/stdc++.h>
#define Nmax 100005
using namespace std;
ifstream f("cerere.in");
ofstream g("cerere.out");
int N, nr;
int pos[Nmax], ans[Nmax], st[Nmax];
vector<int> G[Nmax];
void DFS(int node, int father) {
++nr, st[nr] = node;
if (pos[node]) ans[node] = 1 + ans[st[nr - pos[node]]];
for (auto it: G[node])
if (it != father) DFS(it, node);
--nr;
}
int main()
{
f >> N;
for (int i = 1; i <= N; ++i)
f >> pos[i];
for (int i = 1; i < N; ++i) {
int x, y;
f >> x >> y;
G[x].push_back(y);
}
for (int i = 1; i <= N; ++i)
if (pos[i] == 0) {
DFS(i, 0);
break;
}
for (int i = 1; i <= N; ++i)
g << ans[i] << " ";
return 0;
}