2015年ACM/ICPC长春赛区 J题（01字典树）

题目链接：

https://icpcarchive.ecs.baylor.edu/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=5204

 

题意：

给你n(n<=1000)个数，让你选三个数a[i],a[j],a[k](i!=j&&i!=k&&j!=k)，使得(a[i]+a[j])^a[k]最大。(a[i]<=1e9)

思路：01字典树。为了保证i,j,k不相同，我们val[u]>0表示字典树的u节点对应的那个数存在。查询的时候，枚举i,j，删掉a[i],a[j],按val[ch[u][id^1]]是否存在贪心往下走即可。(id表示(a[i]+a[j])的某一位是1还是0)

代码：

#include<bits/stdc++.h>
#define ll long long
#define inf 0x3f3f3f3f;
using namespace std;
const int maxn=1000010;
const int ssize=2;
int n,a[maxn];
struct Trie
{
    int ch[maxn][ssize];
    int sz;
    int val[maxn];
    void init()
    {
        memset(ch[0],0,sizeof(ch[0]));
        sz=1;
        val[0]=0;
    }
    void insert(int s,int v)
    {
        int u=0,id;
        for(int i=31;i>=0;i--)
        {
            if(s&(1<<i)) id=1;
            else id=0;
            if(!ch[u][id])
            {
                ch[u][id]=sz;
                memset(ch[sz],0,sizeof(ch[sz]));
                val[sz++]=0;
            }
            u=ch[u][id];
            val[u]+=v;
        }
        //val[u]+=v;
    }
    int find(int s)
    {
        int u=0,id,res=0;
        for(int i=31;i>=0;i--)
        {
            res<<=1;
            if(s&(1<<i)) id=1;
            else id=0;
            if(val[ch[u][id^1]]) {
                //cout<<s<<" "<<i<<" "<<u<<endl;
                u=ch[u][id^1];res++;}
            else u=ch[u][id];
        }
        //cout<<res<<endl;
        return res;
    }
}tr;
void add(int s)
{
    tr.insert(s,1);
}
void del(int s)
{
    tr.insert(s,-1);
}
int main()
{
    int T,cas=1,ans;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        tr.init();ans=-inf;
        for(int i=0;i<n;i++)
        {
        scanf("%d",&a[i]);
        add(a[i]);
        }
        for(int i=0;i<n;i++)
        for(int j=i+1;j<n;j++)
        {
            del(a[i]);
            del(a[j]);
            ans=max(ans,tr.find((a[i]+a[j])));
            add(a[i]);
            add(a[j]);
        }
        printf("%d\n",ans);
    }
    return 0;
}