Advertisement
ivnikkk

Untitled

May 22nd, 2022 (edited)
70
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 4.71 KB | None | 0 0
  1. #define _CRT_SECURE_NO_WARNINGS
  2. #define debug(l) cerr<<#l<<' '<<l<<'\n';
  3. #include "bits/stdc++.h"
  4. using namespace std;
  5. #define all(a) a.begin(), a.end()
  6. typedef int ll;
  7. typedef pair<ll, ll> pll;
  8. typedef long double ld;
  9. struct Node {
  10.     unordered_map<ll, ll> colors;
  11. };
  12. struct Node_help {
  13.     vector<ll> colors;
  14. };
  15. vector<Node_help> centroid_build_vertex;
  16. vector<Node> centroid_vertex;
  17. vector<ll> clr;
  18. vector<vector<ll>> g;
  19. const ll LG = 17;
  20. vector<vector<ll>> up;
  21. vector<ll> d;
  22.  
  23. void dfs(ll v, ll p) {
  24.     if (p == -1) {
  25.         d[v] = 0;
  26.         up[0][v] = 0;
  27.     }
  28.     else {
  29.         d[v] = d[p] + 1;
  30.     }
  31.     for (ll l = 1; l < LG; l++) {
  32.         up[l][v] = up[l - 1][up[l - 1][v]];
  33.     }
  34.     for (ll& u : g[v]) {
  35.         if (u != p) {
  36.             up[0][u] = v;
  37.             dfs(u, v);
  38.         }
  39.     }
  40. }
  41. ll get_lca(ll a, ll b) {
  42.     if (a == b)return a;
  43.     if (d[b] > d[a])swap(a, b);
  44.     for (ll i = LG - 1; i >= 0; i--) {
  45.         if (d[up[i][a]] >= d[b]) a = up[i][a];
  46.     }
  47.     if (a == b) return a;
  48.     for (ll i = LG - 1; i >= 0; i--) {
  49.         if (up[i][a] != up[i][b]) a = up[i][a], b = up[i][b];
  50.     }
  51.     return up[0][a];
  52. }
  53. ll get_dist(ll a, ll b) {
  54.     ll lca = get_lca(a, b);
  55.     return d[b] + d[a] - 2 * d[lca];
  56. }
  57. struct Centroid_decomposition {
  58.     vector<vector<ll>> gr;
  59.     vector<ll> tree, siz;
  60.     ll root = -1;
  61.     Centroid_decomposition(ll n) {
  62.         tree.resize(n, -2);
  63.         siz.resize(n);
  64.     }
  65.     void sizes(ll v, ll p) {
  66.         siz[v] = 1;
  67.         for (ll& u : g[v]) {
  68.             if (u != p && tree[u] == -2) {
  69.                 sizes(u, v);
  70.                 siz[v] += siz[u];
  71.             }
  72.         }
  73.     }
  74.     ll centroid(ll v, ll p, ll s) {
  75.         for (ll& u : g[v]) {
  76.             if (u != p && tree[u] == -2 && siz[u] > s / 2) {
  77.                 return centroid(u, v, s);
  78.             }
  79.         }
  80.         return v;
  81.     }
  82.     void build(ll v, ll p) {
  83.         sizes(v, -1);
  84.         tree[v] = p;
  85.         if (p != -1) {
  86.             gr[v].push_back(p);
  87.             gr[p].push_back(v);
  88.         }
  89.         centroid_build_vertex[v].colors.push_back({ v });
  90.         for (ll& u : g[v]) {
  91.             if (tree[u]==-2) {
  92.                 build(centroid(u, -1, siz[u]), v);
  93.             }
  94.         }
  95.         for (ll& u : gr[v]) {
  96.             if (u != p) {
  97.                 if ((ll)centroid_build_vertex[v].colors.size() < centroid_build_vertex[u].colors.size()) {
  98.                     centroid_build_vertex[v].colors.swap(centroid_build_vertex[u].colors);
  99.                 }
  100.                 for (ll& i : centroid_build_vertex[u].colors) {
  101.                     centroid_build_vertex[v].colors.push_back(i);
  102.                 }
  103.                 centroid_build_vertex[u].colors.clear();
  104.             }
  105.         }
  106.         gr[v].clear();
  107.         for (ll& i : centroid_build_vertex[v].colors) {
  108.             ll x = get_dist(i, v);
  109.             if (centroid_vertex[v].colors.find(clr[i]) == centroid_vertex[v].colors.end()) {
  110.                 centroid_vertex[v].colors[clr[i]] = x;
  111.                 continue;
  112.             }
  113.             centroid_vertex[v].colors[clr[i]] = min(centroid_vertex[v].colors[clr[i]], x);
  114.         }
  115.     }
  116.     void build_tree() {
  117.         ll v = 0;
  118.         sizes(v, -1);
  119.         gr.resize(tree.size());
  120.         centroid_build_vertex.resize(tree.size());
  121.         centroid_vertex.resize(tree.size());
  122.         build(centroid(v, -1, siz[v]), -1);
  123.         gr.clear();
  124.         siz.clear();
  125.     }
  126. };
  127. signed main() {
  128. #ifdef _DEBUG
  129.     freopen("input.txt", "r", stdin);
  130.     freopen("output.txt", "w", stdout);
  131. #endif
  132.     ios_base::sync_with_stdio(false);
  133.     cin.tie(nullptr);
  134.     cout.tie(nullptr);
  135.     ll n;
  136.     cin >> n;
  137.     g.resize(n);
  138.     clr.resize(n);
  139.     for (ll i = 0; i < n - 1; i++) {
  140.         ll x;
  141.         cin >> x;
  142.         g[x].push_back(i + 1);
  143.         g[i + 1].push_back(x);
  144.     }
  145.     for (ll i = 0; i < n; i++) {
  146.         cin >> clr[i];
  147.     }
  148.     up.resize(LG, vector<ll>(n));
  149.     d.resize(n);
  150.     dfs(0, -1);
  151.     centroid_vertex.resize(n);
  152.     centroid_build_vertex.resize(n);
  153.     Centroid_decomposition U(n);
  154.     U.build_tree();
  155.     g.clear();
  156.     centroid_build_vertex.clear();
  157.     ll q;
  158.     cin >> q;
  159.     while (q--) {
  160.         ll v, c;
  161.         cin >> v >> c;
  162.         ll it = v;
  163.         ll ans = INT_MAX;
  164.         while (it != -1) {
  165.             auto& clrs = centroid_vertex[it].colors;
  166.             auto it_c = clrs.find(c);
  167.             if (it_c != clrs.end()) {
  168.                 ans = min(ans, (it_c->second) + get_dist(v, it));
  169.             }
  170.             it = U.tree[it];
  171.         }
  172.         if (ans == INT_MAX) {
  173.             cout << -1 << ' ';
  174.         }
  175.         else {
  176.             cout << ans << ' ';
  177.         }
  178.     }
  179. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement