Pagini recente » Cod sursa (job #2563960) | Cod sursa (job #1062859) | Cod sursa (job #2806700) | Cod sursa (job #2751086) | Cod sursa (job #2957239)
#include<fstream>
#include<vector>
#define NMAX 100000
#define LOG 18
std::ifstream cin("cerere.in");
std::ofstream cout("cerere.out");
using namespace std;
int n, u, v, root;
vector<int>g[NMAX + 1], ancestor, viz, dp;
int up[NMAX + 1][LOG];
int binary_lifting(int node, int k) {
for (int i = LOG - 1; i >= 0; --i)
if (k & (1 << i)) node = up[node][i];
return node;
}
void dfs(int node = root, int parent = 0) {
for (auto i : g[node]) {
if (i ^ parent) {
if (!ancestor[i]) dp[i] = 0;
else dp[i] = 1 + dp[binary_lifting(i, ancestor[i])];
dfs(i, node);
}
}
}
int main() {
cin >> n;
ancestor = viz = dp = vector<int>(n + 1);
for (int i = 1; i <= n; i++) cin >> ancestor[i];
for (int i = 1; i < n; ++i) {
cin >> u >> v;
g[u].push_back(v);
up[v][0] = u;
viz[v] = 1;
}
for (int i = 1; i <= n; ++i)
for (int j = 1; j < LOG; ++j)
up[i][j] = up[up[i][j - 1]][j - 1];
for (int i = 1; i <= n; i++)
if (!viz[i]) root = i;
dfs();
for (int i = 1; i <= n; i++) cout << dp[i] << " ";
return 0;
}