NEFU 1270 智力异或（2）|| 2017icpc青岛站 热身赛 C （01字典树）

题意：

中文

思路：

将数插入到01字典树中并记录子树的大小和子树中所有数的位信息。

查询时在字典树中搜索前K个即可。

注意数是可以重复的

代码：

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
const int MAXN=1e5+7;
int n,m,p;
int ls[MAXN*35]={0},rs[MAXN*35]={0},num[MAXN*35]={0},bit[MAXN*35][32]={0};
int cnt=0;
long long two[32];
int trans(int rt){
    if(rt==0){
        ++cnt;
        return cnt;
    }
    return rt;
}
void ini(){
    for(int i=0;i<=cnt;i++){
        ls[i]=rs[i]=num[i]=0;
        for(int j=0;j<32;j++) bit[i][j]=0;
    }
    cnt=1;p=0;
}
void insert(int x){
    int rt=1;
    num[rt]++;
    for(int i=31;i>=0;i--) bit[rt][i]+=(x>>i)%2;
    for(int i=31;i>=0;i--){
        int c=(x>>i)%2;
        if(c==0){
            ls[rt]=trans(ls[rt]);
            rt=ls[rt];
            num[rt]++;
            for(int j=31;j>=0;j--) bit[rt][j]+=(x>>j)%2;
        }else{
            rs[rt]=trans(rs[rt]);
            rt=rs[rt];
            num[rt]++;
            for(int j=31;j>=0;j--) bit[rt][j]+=(x>>j)%2;
        }
    }
}

long long query(int x,int k){
    int rt=1;
    long long ans=0;
    for(int i=31;i>=0&&k;i--){
        int c=(x>>i)%2;
        if(c==0){
            if(rs[rt]){
                if(num[rs[rt]]<=k){
                    k-=num[rs[rt]];
                    for(int j=31;j>=0;j--){
                        int p=(x>>j)%2;
                        if(p==0){
                            ans+=bit[rs[rt]][j]*two[j];
                        }else{
                            ans+=(num[rs[rt]]-bit[rs[rt]][j])*two[j];
                        }
                    }
                    rt=ls[rt];
                }else{
                    rt=rs[rt];
                }
            }else{
                rt=ls[rt];
            }
        }else{
            if(ls[rt]){
                if(num[ls[rt]]<=k){
                    k-=num[ls[rt]];
                    for(int j=31;j>=0;j--){
                        int p=(x>>j)%2;
                        if(p==0){
                            ans+=bit[ls[rt]][j]*two[j];
                        }else{
                            ans+=(num[ls[rt]]-bit[ls[rt]][j])*two[j];
                        }
                    }
                    rt=rs[rt];
                }else{
                    rt=ls[rt];
                }
            }else{
                rt=rs[rt];
            }
        }
    }
    if(k){
        for(int i=31;i>=0;i--){
            int p=(x>>i)%2;
            if((p==0&&bit[rt][i])||(p==1&&!bit[rt][i])){
                ans+=k*two[i];
            }
        }
    }
    return ans;
}
int main(){
    int T;
    //freopen("data.in","r",stdin);
    //freopen("out.txt","w",stdout);
    two[0]=1;
    for(int i=1;i<32;i++) two[i]=two[i-1]*2;
    scanf("%d",&T);
    while(T--){
        scanf("%d%d",&n,&m);
        ini();
        for(int i=1;i<=n;i++){int x;scanf("%d",&x);
            insert(x);
        }
        while(m--){int op,x;scanf("%d%d",&op,&x);
            if(op==1){
                p^=x;
            }else{
                printf("%lld\n",query(p,x));
            }
        }
    }
}