Advertisement
Dmaxiya

最大异或结点 参考代码

Mar 9th, 2025 (edited)
180
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.96 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. typedef long long LL;
  5. const int maxn = 100000 + 100;
  6. const int maxcnt = 3200000 + 100;
  7. const int Size = 2;
  8.  
  9. struct Trie {
  10.     int root, cnt;
  11.     int tree[maxcnt][Size];
  12.     int val[maxcnt];
  13.  
  14.     int create() {
  15.         ++cnt;
  16.         memset(tree[cnt], 0, sizeof(tree[cnt]));
  17.         val[cnt] = 0;
  18.         return cnt;
  19.     }
  20.  
  21.     void Init() {
  22.         cnt = 0;
  23.         root = create();
  24.     }
  25.  
  26.     inline int id(int x, int dig) {
  27.         return (x >> dig) & 1;
  28.     }
  29.  
  30.     void add(int x, int d) {
  31.         int pos = root;
  32.         for(int i = 30; i >= 0; --i) {
  33.             int w = id(x, i);
  34.             if(tree[pos][w] == 0) {
  35.                 tree[pos][w] = create();
  36.             }
  37.             pos = tree[pos][w];
  38.             val[pos] += d;
  39.         }
  40.     }
  41.  
  42.     int query(int x) {
  43.         int pos = root;
  44.         int ret = 0;
  45.         for(int i = 30; i >= 0; --i) {
  46.             int w = id(x, i);
  47.             if (tree[pos][w ^ 1] != 0 && val[tree[pos][w ^ 1]] != 0) {
  48.                 ret |= (1 << i);
  49.                 pos = tree[pos][w ^ 1];
  50.             } else {
  51.                 pos = tree[pos][w];
  52.             }
  53.         }
  54.         return ret;
  55.     }
  56. };
  57.  
  58. int n, fa, ans;
  59. int num[maxn];
  60. vector<int> G[maxn];
  61. Trie trie;
  62.  
  63. int main() {
  64. #ifdef ExRoc
  65.     freopen("test.txt", "r", stdin);
  66. #endif
  67.     ios::sync_with_stdio(false);
  68.  
  69.     cin >> n;
  70.     trie.Init();
  71.     for (int i = 0; i < n; ++i) {
  72.         cin >> num[i];
  73.         trie.add(num[i], 1);
  74.     }
  75.     for (int i = 0; i < n; ++i) {
  76.         cin >> fa;
  77.         if (fa != -1) {
  78.             G[fa].push_back(i);
  79.             G[i].push_back(fa);
  80.         }
  81.     }
  82.     for (int i = 0; i < n; ++i) {
  83.         for (int pos : G[i]) {
  84.             trie.add(num[pos], -1);
  85.         }
  86.         ans = max(ans, trie.query(num[i]));
  87.         for (int pos : G[i]) {
  88.             trie.add(num[pos], 1);
  89.         }
  90.     }
  91.     cout << ans << endl;
  92.  
  93.     return 0;
  94. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement