Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #define _CRT_SECURE_NO_WARNINGS
- #define debug(l) cerr<<#l<<' '<<l<<'\n';
- #include "bits/stdc++.h"
- using namespace std;
- #define all(a) a.begin(), a.end()
- typedef int ll;
- typedef pair<ll, ll> pll;
- typedef long double ld;
- // вроде не особо должны эти массивы с этими структурами много занимать т к глубина дерева центроидов <=log2(n)
- struct Node {
- // в массиве эта структура будет лежать в i позиции -> значит эта структура будет хранить значение для поддерева i
- // ключ - цвет
- // значение мультисет расстояний от детей в центроидном дереве до вершин с цветом ключа
- unordered_map<ll, multiset<ll>> colors;
- };
- struct Node_help {
- // в массиве эта структура будет лежать в i позиции -> значит эта структура будет хранить значение для поддерева i
- // просто все вершины поддерева i pair<> {значение цвета, номер вершины(лежащая в поддереве i) }
- vector<pll> colors;
- };
- vector<Node_help> centroid_build_vertex;
- vector<Node> centroid_vertex;
- vector<ll> clr;
- vector<vector<ll>> g;
- const ll LG = 17;
- vector<vector<ll>> up;
- vector<ll> d;
- void dfs(ll v, ll p) {
- if (p == -1) {
- d[v] = 0;
- up[0][v] = 0;
- }
- else {
- d[v] = d[p] + 1;
- }
- for (ll l = 1; l < LG; l++) {
- up[l][v] = up[l - 1][up[l - 1][v]];
- }
- for (ll& u : g[v]) {
- if (u != p) {
- up[0][u] = v;
- dfs(u, v);
- }
- }
- }
- ll get_lca(ll a, ll b) {
- if (a == b)return a;
- if (d[b] > d[a])swap(a, b);
- for (ll i = LG - 1; i >= 0; i--) {
- if (d[up[i][a]] >= d[b]) a = up[i][a];
- }
- if (a == b) return a;
- for (ll i = LG - 1; i >= 0; i--) {
- if (up[i][a] != up[i][b]) a = up[i][a], b = up[i][b];
- }
- return up[0][a];
- }
- ll get_dist(ll a, ll b) {
- ll lca = get_lca(a, b);
- return d[b] + d[a] - 2 * d[lca];
- }
- struct Centroid_decomposition {
- vector<vector<ll>> gr;
- vector<ll> tree, siz;
- vector<bool> used;
- ll root = -1;
- Centroid_decomposition(ll n) {
- tree.resize(n);
- siz.resize(n);
- used.resize(n);
- }
- void sizes(ll v, ll p) {
- siz[v] = 1;
- for (ll& u : g[v]) {
- if (u != p && !used[u]) {
- sizes(u, v);
- siz[v] += siz[u];
- }
- }
- }
- ll centroid(ll v, ll p, ll s) {
- for (ll& u : g[v]) {
- if (u != p && !used[u] && siz[u] > s / 2) {
- return centroid(u, v, s);
- }
- }
- return v;
- }
- void build(ll v, ll p) {
- sizes(v, -1);
- used[v] = 1;
- tree[v] = p;
- if (p != -1) {
- gr[v].push_back(p);
- gr[p].push_back(v);
- }
- centroid_build_vertex[v].colors.push_back({ clr[v],v });
- for (ll& u : g[v]) {
- if (!used[u]) {
- build(centroid(u, -1, siz[u]), v);
- }
- }
- for (ll& u : gr[v]) {
- if (u != p) {
- if ((ll)centroid_build_vertex[v].colors.size() < centroid_build_vertex[u].colors.size()) {
- centroid_build_vertex[v].colors.swap(centroid_build_vertex[u].colors);
- }
- for (pll& i : centroid_build_vertex[u].colors) {
- centroid_build_vertex[v].colors.push_back(i);
- }
- centroid_build_vertex[u].colors.clear();
- }
- }
- for (pll& i : centroid_build_vertex[v].colors) {
- centroid_vertex[v].colors[i.first].insert(get_dist(i.second, v));
- }
- }
- void build_tree() {
- ll v = 0;
- sizes(v, -1);
- gr.resize(tree.size());
- centroid_build_vertex.resize(tree.size());
- centroid_vertex.resize(tree.size());
- build(centroid(v, -1, siz[v]), -1);
- used.clear();
- siz.clear();
- }
- };
- signed main() {
- #ifdef _DEBUG
- freopen("input.txt", "r", stdin);
- freopen("output.txt", "w", stdout);
- #endif
- ios_base::sync_with_stdio(false);
- cin.tie(nullptr);
- cout.tie(nullptr);
- ll n;
- cin >> n;
- g.resize(n);
- clr.resize(n);
- for (ll i = 0; i < n - 1; i++) {
- ll x;
- cin >> x;
- g[x].push_back(i + 1);
- g[i + 1].push_back(x);
- }
- for (ll i = 0; i < n; i++) {
- cin >> clr[i];
- }
- up.resize(LG, vector<ll>(n));
- d.resize(n);
- dfs(0, -1);
- centroid_vertex.resize(n);
- centroid_build_vertex.resize(n);
- Centroid_decomposition U(n);
- U.build_tree();
- ll q;
- cin >> q;
- while (q--) {
- ll v, c;
- cin >> v >> c;
- ll it = v;
- ll ans = INT_MAX;
- while (it != -1) {
- if (!centroid_vertex[it].colors[c].empty()) {
- auto x = *centroid_vertex[it].colors[c].begin();
- ans = min(ans, x + get_dist(v, it));
- }
- it = U.tree[it];
- }
- if (ans == INT_MAX) {
- cout << -1 << ' ';
- }
- else {
- cout << ans << ' ';
- }
- }
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement