HDU - 5536 Chip Factory 01字典树带删除操作

HDU - 5536 Chip Factory

题意：

给你n个数，从中选出三个数（这三个数的下标各不相同）使 ( s[i]+s[j] )^s[k] 的值最大，输出最大值。

思路：

将输入的数转换为二进制存入字典树中，每次枚举两个数将这两个数删除，从剩余的数中查找 异或 值最大的数，查找完之后再将这两个数插入字典树中。

参考博客：https://blog.csdn.net/hyf20144055065/article/details/77291844

这道题暴力也能过。。。

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int maxn = 1e5+10;
int tr[maxn][2],v[maxn],num[maxn];
int tot,root;
int s[maxn];
void insert(int x)  //插入
{
    root = 0;
    for(int i=31;i>=0;i--) //本来i从32开始减的，c会计算错，（9>>32)&1会算成1
    {
        int k = (x>>i)&1;  //转二进制存储
        if(!tr[root][k])
            tr[root][k] = ++tot;
        root = tr[root][k];
        v[root]++;  //记录数位出现次数
    }
    num[root] = x;  //保存该数  求最大值时用到
}
void dele(int x)  //删除
{
    root = 0;
    for(int i=31;i>=0;i--)
    {
        int k = (x>>i)&1;
        root = tr[root][k];
        v[root]--;  //减去该数位
    }
}
int query(int x)
{
    root = 0;
    for(int i=31;i>=0;i--)
    {
        int k = (x>>i)&1;
        if(tr[root][!k]&&v[tr[root][!k]])  //数位不相同的异或值最大
            root = tr[root][!k];
        else
            root = tr[root][k];
    }
    return x^num[root];
}
int main()
{
    int t,n;
    int ans;
    scanf("%d",&t);
    for(int i=0;i<t;i++)
    {
        memset(v,0,sizeof(v));
        memset(tr,0,sizeof(tr));
        memset(num,0,sizeof(num));
        tot = 0;
        ans = -1;
        
        scanf("%d",&n);
        for(int j=0;j<n;j++)
        {
            scanf("%d",&s[j]);
            insert(s[j]);
        }
        for(int j=0;j<n;j++)
        {
            dele(s[j]);
            for(int k=j+1;k<n;k++)
            {
                dele(s[k]);
                ans = max(ans,query(s[j]+s[k]));
                insert(s[k]);
            }
            insert(s[j]);
        }
        printf("%d\n",ans);
    }
    return 0;
}