【HDOJ5536】Chip Factory（Trie树）

题意：给定n个数字，第i个数字为a[i]，求max((a[i]+a[j])^a[k])，其中i，j，k互不相同

n<=1000,0<=a[i]<=1e9

思路：队友写的，抱大腿

先对于a[i]建立Trie树

枚举i和j后删除Trie中的i和j，然后就是经典的Trie树应用：优先朝与a[i]+a[j]的当前位不同的方向走，能获得最大xor和

#include<bits/stdc++.h>
using namespace std;
#define LL long long  
const int maxn=1e3+5;
int a[maxn*maxn][2];
int b[maxn],num[maxn*maxn*2],tot;
void init()
{
    tot=1;
}
void insert (int x)
{
    int rt=1;
    for(int i=30;i>=0;i--)
    {
        int g=(x>>i)&1;
        if(a[rt][g]==0)
        {
            a[rt][g]=++tot;
        }
        rt=a[rt][g];
        num[rt]++;
    }
}
void del(int x)
{
    int rt=1;
    for(int i=30;i>=0;i--)
    {
        int g=(x>>i)&1;
        rt=a[rt][g];
        num[rt]--;
    }
}
int find(int x)
{
    int rt=1;
    int sum=0;
    for(int i=30;i>=0;i--)
    {
        int g=(x>>i)&1;
    //    printf("%d %d %d
",g,a[rt][!g],a[rt][g]);
        if(a[rt][!g]==0||num[a[rt][!g]]==0)
        {
            rt=a[rt][g];
        }
        else
        {
        //    printf("i = %d
",i);
            rt=a[rt][!g];
            sum+=1<<i;
        }
        //printf("%d
",sum);
    }
    return sum;
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        init();
        int n;
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&b[i]);
            insert(b[i]);
        }
        int ma=0;
        for(int i=1;i<=n;i++)
            for(int j=i+1;j<=n;j++)
            {
                del(b[i]);
                del(b[j]);
            //    printf("%d
",find(b[i]+b[j]));
            //    puts("sdadas");
                ma=max(ma,find(b[i]+b[j]));
                insert(b[i]);
                insert(b[j]);
            }
        for(int i=0;i<=tot;i++)
        {
            a[i][0]=a[i][1]=0;
            num[i]=0;
        }
        printf("%d
",ma);
    }
}