Pagini recente » Cod sursa (job #111732) | Cod sursa (job #3271179) | Cod sursa (job #2295178) | Cod sursa (job #2733678) | Cod sursa (job #3167328)
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
const int NMAX = 100003;
vector<int> g[NMAX];
vector<int> stramosi;
int dp[NMAX], v[NMAX];
bool apare[NMAX];
void dfs(int nod) {
if (v[nod] == 0)
dp[nod] = 0;
else
dp[nod] = dp[stramosi[stramosi.size() - v[nod]]] + 1;
stramosi.push_back(nod);
for (auto nxt : g[nod]) {
dfs(nxt);
}
stramosi.pop_back();
}
int main() {
int n;
fin >> n;
for (int i = 1; i <= n; i++) {
fin >> v[i];
}
for (int i = 1; i <= n - 1; i++) {
int x, y;
fin >> x >> y;
g[x].push_back(y);
apare[y] = true;
}
int s = 0;
for (int i = 1; i <= n; i++) {
if (!apare[i]) {
s = i;
break;
}
}
dfs(s);
for (int i = 1; i <= n; i++) {
fout << dp[i] << ' ';
}
return 0;
}