【HDU】 5536 Chip Factory

Chip Factory

---

• 题目链接
  Chip Factory

---

• 题目大意

说有一组数，你需要从其中选取3个数a b c，使得(a+b) xor c 达到最大。

---

• 题解

建立一棵字典树，每个数用2进制表示插入树中，我们先枚举a跟b，将a跟b从字典树中删除，最后在字典树中查找a+b的“反码”（因为要使最后的异或结果最大）。如果有节点就继续查找，没有节点就走另一边。

---

• 代码

#include <iostream>
#include <cstring>
#include <cstdio>

using namespace std;

struct Trie
{
    Trie *ch[2];
    int val;
};
Trie *root;
int a[1005],T,n,ans;

void _insert(int s)
{
    Trie *p=root,*q;
    for (int i=30;i>=0;i--)
    {
        int ch;
        if (s&(1<<i)) ch=1;
        else ch=0;
        if (p->ch[ch]) p->ch[ch]->val++;
        else
        {
            q=new Trie;
            for (int i=0;i<2;i++) q->ch[i]=NULL;
            q->val=1;
            p->ch[ch]=q;
        }
        p=p->ch[ch];
    }
}

void _delete(int s)
{
    Trie *p=root;
    int ch;
    for (int i=30;i>=0;i--)
    {
        if (s&(1<<i)) ch=1;
        else ch=0;
        if (p->ch[ch]) p->ch[ch]->val--;
        p=p->ch[ch];
    }
}

int solve(int n)
{
    int ans=1;
    for (int i=1;i<=n;i++) ans*=2;
    return ans;
}

int qur(int s)
{
    Trie *p=root;
    int num,ans=0;
    for (int i=30;i>=0;i--)
    {
        if (s&(1<<i)) num=0;
        else num=1;
        if (p->ch[num] && p->ch[num]->val)
        {
            ans+=solve(i);
            p=p->ch[num];
        }
        else if (p->ch[!num]) p=p->ch[!num];
    }
    return ans;
}

void del(Trie *p)
{
    for (int i=0;i<2;i++) if (p->ch[i]) del(p->ch[i]);
    delete []p;
}

int main()
{
    scanf("%d",&T);
    while (T--)
    {
        root=new Trie;
        for (int i=0;i<2;i++) root->ch[i]=NULL;
        ans=0;
        scanf("%d",&n);
        for (int i=0;i<n;i++)
        {
            scanf("%d",&a[i]);
            _insert(a[i]);
        }
        for (int i=0;i<n-1;i++)
        {
            _delete(a[i]);
            for (int j=i+1;j<n;j++)
            {
                _delete(a[j]);
                ans=max(ans,qur(a[i]+a[j]));
                _insert(a[j]);
            }
            _insert(a[i]);
        }
        printf("%d\n",ans);
        del(root);
    }
    return 0;
}