Cod sursa(job #2957072)

Utilizator apocal1ps13Stefan Oprea Antoniu apocal1ps13 Data 21 decembrie 2022 17:14:42
Problema Cerere Scor 85
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.11 kb
#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;
int up[NMAX + 1][LOG];
void dfs(int node = root, int parent = 0) {
	for (auto i : g[node]) {
		if (i ^ parent) {
			up[i][0] = node;
			for (int j = 1; j < LOG; ++j)
				up[i][j] = up[up[i][j - 1]][j - 1];
			dfs(i, node);
		}
	}
}
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;
}
int main() {
	cin >> n;
	ancestor = viz = 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);
		viz[v] = 1;
	}
	for (int i = 1; i <= n; i++)
		if (!viz[i]) root = i;
	dfs();
	for (int i = 1; i <= n; i++) {
		int ki = ancestor[i];
		int new_ancestor = i;
		int cnt = 0;
		while (ki) {
			cnt++;
			new_ancestor = binary_lifting(new_ancestor, ki);
			ki = ancestor[new_ancestor];
		}
		cout << cnt << " ";
	}
	return 0;
}