Vasiliy's Multiset CodeForces -706D || 01字典树模板

就是一个模板

注意这题有一个要求：有一个额外的0一直保持在集合中

#include<cstdio>
#include<algorithm>
using namespace std;
const int _LEN=30;
//上限为2^30-1，即二进制最多30位
int lft[40];
namespace Trie
{
    int ch[7001000][2],sz[7001000];
    int mem;
    void insert(int x,int cur)
    {
        int i;bool t;
        for(i=_LEN-1;i>=0;--i)
        {
            t=x&lft[i];
            if(!ch[cur][t])    ch[cur][t]=++mem;
            ++sz[cur];cur=ch[cur][t];
        }
        ++sz[cur];
    }
    int query(int x,int cur)
    {
        int i,ans=0;bool t;
        for(i=_LEN-1;i>=0;--i)
        {
            t=x&lft[i];
            if(sz[ch[cur][t^1]])    ans|=lft[i],cur=ch[cur][t^1];
            else    cur=ch[cur][t];
        }
        return ans;
    }
    void erase(int x,int cur)
    {
        int i;bool t;
        for(i=_LEN-1;i>=0;--i)
        {
            t=x&lft[i];
            --sz[cur];cur=ch[cur][t];
        }
        --sz[cur];
    }
};
int root;
char ss[10];
int main()
{
    int i,q,x;
    lft[0]=1;
    for(i=1;i<=_LEN;i++)    lft[i]=lft[i-1]<<1;
    root=++Trie::mem;Trie::insert(0,root);
    scanf("%d",&q);
    while(q--)
    {
        scanf("%s%d",ss,&x);
        if(ss[0]=='+')
        {
            Trie::insert(x,root);
        }
        else if(ss[0]=='-')
        {
            Trie::erase(x,root);
        }
        else if(ss[0]=='?')
        {
            printf("%d
",Trie::query(x,root));
        }
    }
    return 0;
}