hdu5536Chip Factory 字典树与异或 2015ACM/ICPC亚洲区长春站现场赛

题意： 给定n个个数s1, s2... sn，求 maxi,j,k(si+sj)⊕sk      

重点是满足条件： i ≠ j ≠ k。

思路：把每个数都插入字典树，枚举两个数的和。考虑到可能会有重复的数，所以我们定义flag来存放下标，重复的只存放第一个下标。

假设当前查询的两个数的下标是I, J，查询到的结果最大值的结点存放的下标是K。

如果K ≠ I 且 K ≠J，则满足条件。

如果（K = I 或者 K = J） 且 选择的两个数其值相同，那么，如果这个数在整个数组中存在2个以上，则这3个数满足要求。

如果（K = I 或者 K = J ）且 选择的两个数其值不同，那么值为S[K]的数存在1个以上，则满足条件。

否则不满足条件 i ≠ j ≠ k。

①不满足条件我们就需要重新查询，那么想想，最有可能符合条件且值最大的结点会在哪里？

这个结点一定是在从之前的查询中的那条链上的点，且这个点有两个儿子，且它离根最远。查询的时候我们记录这个点。

继续查询，找到这个点，我们刚才肯定是往能得到最大值的点的方向走的，这个时候我们在这个时候往另一个方向走。

如果这个时候又找到了一个下标等于I 或者等于 J的点，即不满足条件，那么我们返回到①处，这次执行肯定能找到满足条件的点。

代码（解释结合代码一起看吧）：

#include<stdio.h>
#include<string.h>
#include<algorithm>

using namespace std;

const int N = 1024;

int a[N];
int twochild[2];

struct Trie
{
    int next[2];
    int flag;
    int cnt;
    void init()
    {
        next[0] = next[1] = -1;
        flag = -1;
        cnt = 0;
    }
};

Trie trie[N<<6];
int sz;

void init()
{
    sz = 1;
    trie[0].init();
}

void myinsert(int n, int t1)
{
    int p = 0;
    for(int i = 30; i >= 0; i--)
    {
        int ch = n&(1<<i);
        if(ch) ch = 1;
        if(trie[p].next[ch] == -1)
        {
            trie[p].next[ch] = sz;
            trie[sz++].init();
        }
        p = trie[p].next[ch];
    }
    int cnt = trie[p].cnt;
    if(cnt < 1)
        trie[p].flag = t1;
    trie[p].cnt++;
}

int myquery(int n, int t1, int t2, int seq)
{
    int p = 0;
    twochild[seq] = -1;
    int ret = 0;
    int j = -1;
    for(int i = 30; i >= 0; i--)
    {
        int ch = n & (1 << i);
        if(ch) ch = 1;
        for(j = seq-1; j >= 0; j--){
            if(p == twochild[j])
                break;
        }

        if(j >= 0)
        {
            p = trie[p].next[ch];
            continue;
        }

        if(trie[p].next[0] != -1 && trie[p].next[1] != -1){
            twochild[seq] = p;
        }
        if(trie[p].next[!ch] != -1)
        {
            ret |= (1<<i);
            p = trie[p].next[!ch];
        }
        else p = trie[p].next[ch];
    }
    /*if((trie[p].flag[0] != -1 && trie[p].flag[0] != t1 && trie[p].flag[0] != t2) ||
            (trie[p].flag[1] != -1 && trie[p].flag[1] != t1 && trie[p].flag[1] != t2) || trie[p].cnt > 2)
        return ret;*/
     if(t1 != trie[p].flag && t2 != trie[p].flag)
        return ret;
    if(a[t1] != a[t2] && trie[p].cnt > 1)
        return ret;
    if(a[t1] == a[t2] && trie[p].cnt > 2)
        return ret;

    return -1;
}

int main()
{
    int t;
    scanf("%d", &t);
    while(t--)
    {
        int n;
        init();
        scanf("%d", &n);
        for(int i = 0; i < n; i++)
        {
            scanf("%d", &a[i]);
            myinsert(a[i], i);
        }
        int ans = 0;
        for(int i = 0; i < n; i++)
            for(int j = i+1; j < n; j++)
            {
                int tmp = myquery(a[i]+a[j], i, j, 0);
                if(tmp == -1) tmp = myquery(a[i]+a[j], i, j, 1);
                if(tmp == -1) tmp = myquery(a[i]+a[j], i, j, 2);
                ans = max(ans, tmp);
            }
        printf("%d\n", ans);
    }
    return 0;
}

/*
10
5
19 42 82 84 16
5
29 2 41 18 13
*/