three arrays(hdu多校联测round5)

题目链接

对A和B分别建两棵trie树, 两边同时找, 尽量找相同的, 找到叶子节点时记录答案, 最后将答案排序输出.

#include<bits/stdc++.h>
using namespace std;
int read(){
    int x=0,f=1;char ch=getchar();
    for(;!isdigit(ch);ch=getchar())if(ch=='-')f=-1;
    for(;isdigit(ch);ch=getchar())x=x*10+ch-'0';
    return x*f;
}
const int N=3.5e6+28;
struct TRIE{
    int ch[N][2],sum[N],val[N],tot;
    int Ins(int x){
    int p=1;
    for(int i=31;i>=0;i--){
        int v=((x&(1<<i))!=0);
        if(!ch[p][v])ch[p][v]=++tot;
        p=ch[p][v];
    }
    sum[p]++;
    val[p]=x;
    }
    void Updata(int p){
    sum[p]=sum[ch[p][0]]+sum[ch[p][1]];
    if(sum[ch[p][0]]<=0)ch[p][0]=0;
    if(sum[ch[p][1]]<=0)ch[p][1]=0;
    }
    void Dfs(int p=1){
    if(val[p]!=-1)return;
    if(ch[p][0])Dfs(ch[p][0]);
    if(ch[p][1])Dfs(ch[p][1]);
    Updata(p);
    }
    void format(){
    for(int i=0;i<=tot;i++){
        sum[i]=0;
        val[i]=-1;
        ch[i][0]=ch[i][1]=0;
    }
    tot=1;
    }
    stack<int>stk;
    int Query(int x){
    int p=1;
    stk.push(p);
    for(int i=31;i>=0;i--){
        int v=((x&(1<<i))!=0);
        if(ch[p][v])p=ch[p][v];
        else p=ch[p][v^1];
        stk.push(p);
    }
    stk.pop();
    sum[p]--;
    while(stk.size())Updata(stk.top()),stk.pop();
    //printf("%d^%d=%d sum=%d\n",val[p],x,val[p]^x,sum[p]);
    return val[p]^x;
    }
    TRIE(){
    memset(sum,0,sizeof(sum));
    memset(val,-1,sizeof(val));
    memset(ch,0,sizeof(ch));
    tot=1;
    }
}A,B;
int ans[N],cnt;
void format(){
    A.format();
    B.format();
    cnt=0;
}
void dfs(int pa=1,int pb=1){
    if(A.val[pa]!=-1){
    ans[++cnt]=A.val[pa]^B.val[pb];
    A.sum[pa]--,B.sum[pb]--;
    return;
    }
    while(A.ch[pa][0]&&B.ch[pb][0]){
    dfs(A.ch[pa][0],B.ch[pb][0]);
    A.Updata(pa),B.Updata(pb);
    }
    while(A.ch[pa][1]&&B.ch[pb][1]){
    dfs(A.ch[pa][1],B.ch[pb][1]);
    A.Updata(pa),B.Updata(pb);
    }
    while(A.ch[pa][0]&&B.ch[pb][1]){
    dfs(A.ch[pa][0],B.ch[pb][1]);
    A.Updata(pa),B.Updata(pb);
    }
    while(A.ch[pa][1]&&B.ch[pb][0]){
    dfs(A.ch[pa][1],B.ch[pb][0]);
    A.Updata(pa),B.Updata(pb);
    }
}
signed main(){
    int t=read();
    while(t--){
    format();
    int n=read();
    for(int i=1;i<=n;i++){
        int a=read();
        A.Ins(a);
    }
    A.Dfs();
    for(int i=1;i<=n;i++){
        int b=read();
        B.Ins(b);
    }
    B.Dfs();
    dfs();
    sort(ans+1,ans+n+1);
    for(int i=1;i<=n;i++){
        printf("%d",ans[i]);
        if(i!=n)putchar(' ');
    }
    puts("");
    }
    return 0;
}