2015年ACM/ICPC长春赛区 J题(01字典树)
题目链接:
题意:
给你n(n<=1000)个数,让你选三个数a[i],a[j],a[k](i!=j&&i!=k&&j!=k),使得(a[i]+a[j])^a[k]最大。(a[i]<=1e9)
思路:01字典树。为了保证i,j,k不相同,我们val[u]>0表示字典树的u节点对应的那个数存在。查询的时候,枚举i,j,删掉a[i],a[j],按val[ch[u][id^1]]是否存在贪心往下走即可。(id表示(a[i]+a[j])的某一位是1还是0)
代码:
#include<bits/stdc++.h>
#define ll long long
#define inf 0x3f3f3f3f;
using namespace std;
const int maxn=1000010;
const int ssize=2;
int n,a[maxn];
struct Trie
{
int ch[maxn][ssize];
int sz;
int val[maxn];
void init()
{
memset(ch[0],0,sizeof(ch[0]));
sz=1;
val[0]=0;
}
void insert(int s,int v)
{
int u=0,id;
for(int i=31;i>=0;i--)
{
if(s&(1<<i)) id=1;
else id=0;
if(!ch[u][id])
{
ch[u][id]=sz;
memset(ch[sz],0,sizeof(ch[sz]));
val[sz++]=0;
}
u=ch[u][id];
val[u]+=v;
}
//val[u]+=v;
}
int find(int s)
{
int u=0,id,res=0;
for(int i=31;i>=0;i--)
{
res<<=1;
if(s&(1<<i)) id=1;
else id=0;
if(val[ch[u][id^1]]) {
//cout<<s<<" "<<i<<" "<<u<<endl;
u=ch[u][id^1];res++;}
else u=ch[u][id];
}
//cout<<res<<endl;
return res;
}
}tr;
void add(int s)
{
tr.insert(s,1);
}
void del(int s)
{
tr.insert(s,-1);
}
int main()
{
int T,cas=1,ans;
scanf("%d",&T);
while(T--)
{
scanf("%d",&n);
tr.init();ans=-inf;
for(int i=0;i<n;i++)
{
scanf("%d",&a[i]);
add(a[i]);
}
for(int i=0;i<n;i++)
for(int j=i+1;j<n;j++)
{
del(a[i]);
del(a[j]);
ans=max(ans,tr.find((a[i]+a[j])));
add(a[i]);
add(a[j]);
}
printf("%d\n",ans);
}
return 0;
}