Codeforces #367 (Div. 2) D. Vasiliy's Multiset （trie 树）

 

 

http://codeforces.com/group/1EzrFFyOc0/contest/706/problem/D 

 

题目：就是有3种操作

+ x向集合里添加 x

- x 删除x元素,(保证存在

? x 查询 x |  集合中元素的最大值

 

思路：就是利用字典树，从高位到低位进行贪心。 比如说给一个数 x=3  ， 对x 各位取反（二进制）（x=~x )，

于是就是 1-0-0-1； 拿 1-0-0-1，从左到右（从高位到地位）顺序，来在字典树中寻找。如果能找到（if )，就接着找下去；

如果找不到（else），就退而求其次（贪心），找另外一个分岔点。

在构造的时候从32或者31开始，大于1e9，相当于0-0-0-0-0-0-0-0-0-0-0-0-1-0-1-1 这样 ，在树中 前面多几个0没关系。

 

不过数组要尽量开大一点（尽管我不知道具体应该开多少）

 

 

#include<iostream>
#include<cstdio>
#include <cctype>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<string>
#include<cmath>
#include<set>
#include<vector>
#include<stack>
#include<queue>
#include<map>
using namespace std;
#define ll long long
#define mem(a,x) memset(a,x,sizeof(a))
#define se second
#define fi first
const int INF= 0x3f3f3f3f;
const int N=4e6+5;

int n,cnt=1;
int trie[N][2]={0},num[N]={0}; 

void insert(int c,ll x)
{
    int rt=0;
    for(int i=31;i>=0;i--)
    {
        int k=(x>>i)&1; //二进制高位到低位的 0 1情况
        if(!trie[rt][k]) 
            trie[rt][k]=cnt, cnt++;
        rt=trie[rt][k];
        
        num[rt]+=c; 
    }
}

ll find(ll x)
{
    x=~x;
    int rt=0;
    ll ans=0;
    for(int i=31;i>=0;i--)
    {
        int k=(x>>i)&1;
        ans=ans<<1;
        if( num[trie[rt][k] ] && trie[rt][k] )
        {
            ans++;
            rt=trie[rt][k];
        }
        else
            rt=trie[rt][1-k];  //贪心 
    }
    return ans;
} 
int main()
{
    cin>>n;
    char c;
    ll x;
    
    insert(1,0);
    for(int i=1;i<=n;i++)
    {
        cin>>c>>x;
        if(c=='+')
            insert(1,x);
        else if(c=='-')
            insert(-1,x);
        else
            printf("%lld\n",find(x));
        //cout<<c<<x<<endl;
    }
}