MediaNet_TriesOnGraph.cpp

#include <bits/stdc++.h>
#define ll long long
#define endl '\n'
#define loop(i,s,e) for(ll i = s; s<e ? i < e : i >= e; s<e ? i++ : i--)
#define forrr(i,s,e) for(ll i = s; i >= e; i--)
#define forr(i,s,e) for(ll i = s; i < e; i++)
#define vl vector<ll>
#define vvl vector<vl>
#define pll pair<ll, ll>
#define vp vector<pll>
#define ss second
#define ff first
#define pb push_back
#define pf push_front
#define all(a) a.begin(), a.end()
#define Rand(arr, n) generate_n(arr.begin(), n, random)
#define fastio() ios_base::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL)
#define testt ll t = 1; cin>>t; while(t--)
#ifndef ONLINE_JUDGE
#define debug(x) cerr << #x <<" "; _print(x); cerr << endl;
#else
#define debug(x) 
#endif
using namespace std;

template<class T> void _print(T t) {cerr << t;}
template <class T, class V> void _print(pair <T, V> p) {cerr << "{"; _print(p.ff); cerr << ","; _print(p.ss); cerr << "}";}
template <class T> void _print(vector <T> v) {cerr << "[ "; for (T i : v) {_print(i); cerr << " ";} cerr << "]";}
template <class T> void _print(priority_queue<T> pq) {cerr << "[ "; vector<T> v; while(!pq.empty()){ v.push_back(pq.top()); pq.pop();} for (auto i : v) {_print(i); cerr << " ";} cerr << "]";}
template <class T> void _print(priority_queue<T, vector<T>, greater<T>> pq) {cerr << "[ "; vector<T> v; while(!pq.empty()){ v.push_back(pq.top()); pq.pop();} for (auto i : v) {_print(i); cerr << " ";} cerr << "]";}
template <class T> void _print(set <T> st) {cerr << "[ "; for (auto it = st.begin(); it != st.end(); it++) {_print(*it); cerr << " ";} cerr << "]";}
template <class T> void _print(multiset <T> st) {cerr << "[ "; for (auto it = st.begin(); it != st.end(); it++) {_print(*it); cerr << " ";} cerr << "]";}
template <class T, class V> void _print(map <T, V> mp) {cerr << "[ "; for (auto it = mp.begin(); it != mp.end(); it++) {_print(*it); cerr << " ";} cerr << "]";}
template <class T, class V> void _print(multimap <T, V> mp) {cerr << "[ "; for (auto it = mp.begin(); it != mp.end(); it++) {_print(*it); cerr << " ";} cerr << "]";}

template<typename F, typename S> ostream& operator <<(ostream& ostream, pair<F, S>& p) { cout << p.first << " " << p.second<<" "; return ostream; }
template<typename T> ostream& operator <<(ostream& ostream, vector<T>& v) { for(auto& element : v) cout << element << " "; return ostream;}
template<typename T> ostream& operator <<(ostream& ostream, vector<vector<T>>& v) {for(auto& row : v){ for(auto& cell : row) cout << cell << " "; cout << "\n";} return ostream;}
template<typename F, typename S> istream& operator >>(istream& istream, pair<F, S>& p) { cin >> p.first >> p.second; return istream;}
template<typename T> istream& operator >>(istream& istream, vector<T>& v) { for(auto& element : v) cin >> element; return istream; }

//vl prime(1000001);

int random(int s, int e){return s + rand() % (e - s + 1);}
int flip(int n){ forr(i,0,31){ int num = 1<<i; if(num > n) break; n = n xor num;} return n; }
ll power(ll x, ll y){ ll res = 1; while (y > 0){ if (y & 1) res = (ll)(res*x); y = y>>1; x = (ll)(x*x); } return res; }
vp countArr(vl &arr, ll n){ vp v1; forr(i,0,n-1){ ll cnt = 1; while(i < n-1 && arr[i] == arr[i+1]){ cnt++, i++; } v1.pb({arr[i], cnt}); } if((n > 1 && arr[n-1] != arr[n-2]) || n == 1) v1.pb({arr[n-1], 1}); return v1;}
ll gcd(ll a, ll b){if(b == 0) return a; return gcd(b, a%b);};
vl factors(ll n){ vl fac; for (ll i = 1; i * i <= n; i++){ if (n % i == 0){ fac.pb(i); if (i * i != n)fac.pb(n / i); } } return fac; }
ll maxPow2(ll n){n |= n >> 1; n |= n >> 2; n |= n >> 4; n |= n >> 8; n |= n >> 16; return (n + 1);}
//vl sieve(){vl primes; for(ll i=0;i<1000001;i++) prime[i]=true; for(ll i=2;i<1000001;i++){ if(!prime[i]) continue; primes.pb(i); for(ll j=2;i*j<1000001;j++) prime[i*j]=false;} return primes;}
vp primeFactors(ll n){ vp v; for (ll j = 2; j <= sqrtl(n); j++){ ll cnt = 0; while(n%j == 0) n /= j, cnt++; if(cnt)v.pb({j,cnt}); } if(n!=1) v.pb({n,1}); return v;}

// ll dx[] = {0, 0, 1, 1, 1, -1, -1, -1};
// ll dy[] = {1, -1, 1, -1, 0, 1, -1, 0};
/*
Max xor query
given d and e in each query
find max of (every node in path of root to d)^e
*/
class Tries{
    public:
    class Node{
        public:
        Node *next[2];
        int cnt = 0;
        Node(){
            cnt = 0;
            forr(i,0,2) next[i] = NULL;
        }
    };

    Node *trie;
    Tries(){
        trie = new Node();
    }

    void insert(int &num){
        Node *it = trie;
        for(int i = 20; i >= 0; i--){
            int val = (num&(1<<i) ? 1 : 0);
            if(it->next[val] == NULL)
                it->next[val] = new Node();
            it->next[val]->cnt += 1;
            it = it->next[val];
        }
    }
    void remove(int &num){
        Node *it = trie;
        for(int i = 20; i >= 0; i--){
            int val = (num&(1<<i) ? 1 : 0);
            it->next[val]->cnt -= 1;
            it = it->next[val];
        }
    }

    int find(int &num){
        Node *it = trie;
        int ans = 0;
        for(int i = 20; i >= 0; i--){
            int val = (num&(1<<i) ? 1 : 0);
            if(it->next[val^1] != NULL && it->next[val^1]->cnt != 0){
                it = it->next[val^1];
                ans += (1<<i);
            }
            else it = it->next[val];
        }
        return ans;
    }
};


vector<int> res;
void dfs(int node, vector<vector<int>> &adj, vector<vector<pair<int, int>>> &qry, Tries &trie){
    trie.insert(node);
    for(auto &i : qry[node]){
        res[i.second] = trie.find(i.first);
    }
    for(auto &i : adj[node]){
        dfs(i, adj, qry, trie);
    }
    trie.remove(node);
}


vector<int> solve(){
    ll A; cin>>A;
    vector<int> B(A-1); cin>>B;
    int C; cin>>C;
    vector<int> D(C), E(C); cin>>D>>E;
    
    vector<vector<int>> adj(A+1);
    vector<vector<pair<int, int>>> qry(A+1);
    for(int i = 0; i < A-1; i++){
        adj[B[i]].push_back(i+2);
    }
    for(int i = 0; i < C; i++){
        qry[D[i]].push_back({E[i],i});
    }

    res = vector<int>(C);
    Tries trie = Tries();
    dfs(1,adj,qry,trie);
    return res;
}

int main(){
fastio(); srand(time(NULL));
testt{ 
    vector<int> ans = solve();
    cout<<ans<<endl;
}
return 0;
}
/* use __lg(number) to get nearest power of 2 -> 8,9,10..15 returns 3, 16-31 returns 4*/