2019hdu暑假多校训练赛第五场1002 three arrays hdu6625（字典树+贪心）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=6625

 

题意：给出两个长度为n的数组a,b，让两个数组中的数字一一匹配做异或运算，形成第三个数组c，就出字典序最小的c数组。

数据范围：T≤1000，1≤N≤1e5

 

思路：对于两个数组分别建一棵字典树，字典树上是对于每个数字的二进制串，从高位到低位建，然后对与两棵字典树一直往下找对应二进制位尽可能相同的数，比如0和0,1和1这样异或起来就是0，当不存在0，0和1，1的时候只能选一个1一个0，因为是从高位往低位找的所以这样形成的数字会尽可能的小，把所有的数一一匹配后对形成的数组排序一下输出即可。

#include <bits/stdc++.h>
using namespace std;
const int N=1e5+5;
struct Trie{
    int nex[2],val;
} tree[2][N*32];
int n,m,t,x,tot[2],ans[N];
void Insert(int x,int op){
    int p=0,f;
    for(int i=29; i>=0; i--){
        f=(x>>i)&1;
        if(!tree[op][p].nex[f]){
            tree[op][p].nex[f]=++tot[op];
            tree[op][tot[op]].nex[0]=tree[op][tot[op]].nex[1]=tree[op][tot[op]].val=0;
            ///开新的节点
        }
        p=tree[op][p].nex[f];
        tree[op][p].val++;
    }
}
int query(){
    int p0=0,p1=0,res=0;
    for(int i=29; i>=0; i--){
        int L1=tree[0][p0].nex[0],R1=tree[0][p0].nex[1];
        int L2=tree[1][p1].nex[0],R2=tree[1][p1].nex[1];
        if(tree[0][L1].val&&tree[1][L2].val){
            p0=L1,p1=L2,tree[0][p0].val--,tree[1][p1].val--;
        }
        else if(tree[0][R1].val&&tree[1][R2].val){
            p0=R1,p1=R2,tree[0][p0].val--,tree[1][p1].val--;
        }
        else if(tree[0][R1].val&&tree[1][L2].val){
            p0=R1,p1=L2,tree[0][p0].val--,tree[1][p1].val--;
            res+=(1<<i);
        }
        else{
            p0=L1,p1=R2,tree[0][p0].val--,tree[1][p1].val--;
            res+=(1<<i);
        }
    }
    return res;
}
int main(){
    scanf("%d",&t);
    while(t--){
        tot[0]=tot[1]=0;///初始化根节点
        tree[0][0].nex[0]=tree[0][0].nex[1]=tree[0][0].val=0;
        tree[1][0].nex[0]=tree[1][0].nex[1]=tree[1][0].val=0;
        scanf("%d",&n);
        for(int i=1; i<=n; i++)scanf("%d",&x),Insert(x,0);
        for(int i=1; i<=n; i++)scanf("%d",&x),Insert(x,1);
        for(int i=1; i<=n; i++)ans[i]=query();
        sort(ans+1,ans+n+1);
        for(int i=1; i<=n; i++)printf("%d%c",ans[i]," \n"[i==n]);
    }
    return 0;
}