HDU 5536 Chip Factory 01字典树

题目链接： http://acm.hdu.edu.cn/showproblem.php?pid=5536

题意：数组中求 i,j,k 最大的(A[i] + A[j]) xor A[k] i != j != k
n^2枚举i，j 同时删除字典树中的A[i] A[j] 之后询问A[i] + A[j] 的最大的A[k] 维护最大异或和

代码：

#include <cstdio>
#include <iostream>
#include <queue>
#include <cstring>
#define sf scanf
#define pf printf
using namespace std;
const int maxn = 1000 + 5;
typedef long long LL;
int ch[32 * maxn][2];
LL value[32 * maxn];
int cnt[32 * maxn];
int node_cnt;

inline void init(){
    node_cnt = 1;
    memset(ch[0],0,sizeof(ch));
}

inline void Insert(LL x,int f){
    int cur = 0;
    for(int i = 32;i >= 0;--i){
        int idx = (x >> i) & 1;
        if(!ch[cur][idx]){
            memset(ch[node_cnt],0,sizeof(ch[node_cnt]));
            ch[cur][idx] = node_cnt;
            cnt[node_cnt] = 0;
            value[node_cnt++] = 0;
        }
        cur = ch[cur][idx];
        cnt[cur] += f;
    }
    value[cur] = x;
}

inline LL Query(LL x){
    int cur = 0;
    for(int i = 32;i >= 0;--i){
        int idx = (x >> i) & 1;
        if(ch[cur][idx ^ 1] && cnt[ch[cur][idx^1]]) cur = ch[cur][idx ^ 1];
        else cur = ch[cur][idx];
    }
    return value[cur];
}
int A[maxn];
int main(){
    int T,n,ans,temp,tmp2;
    sf("%d",&T);
    while( T-- ){
        init();
        ans = 0;
        sf("%d",&n);
        for(int i = 0;i < n;++i) {sf("%d",&A[i]);Insert(A[i],1);}
        for(int i = 0;i < n;++i){
            for(int j = i + 1;j < n;++j){
                Insert(A[i],-1);Insert(A[j],-1);
                temp = ( A[i] + A[j] )^ Query(A[i] + A[j]);
                ans = max(ans,temp);
                Insert(A[i],1);Insert(A[j],1);
            }
        }
        pf("%d\n",ans);
    }
    return 0;
}