Codeforces #888
F. Lisa and the Martians
题意:
给出a1…an,和一个参数k,求i,j,x使得x<2^k而且
(
a
i
⊕
x
)
&
(
a
j
⊕
x
)
(a_i \oplus x)\&(a_j \oplus x)
(ai⊕x)&(aj⊕x)
最大。
思路:
要使得这个式子最大我们首先得选出最优ai和aj。容易发现如果ai和aj相同的话是最好的,如果没有相同的那么就相同的二进制位越多越好(高位优先。)
当时不会求ai和aj,看了题解之后发现是使用01Tire树求出来的。对着01Trie树模板改一改即可。
const int N=2e5+10;
int a[N];
int k;
int nex[N*31][2];
int ac[N*31],cnt;
void insert(int x,int v){
int p=0;
for(int i=k-1;i>=0;i--){
int j=(x>>i)&1;
if(!nex[p][j]) nex[p][j]=++cnt;
p=nex[p][j];
}
ac[p]=v;
}
int find(int x){
int p=0;
for(int i=k-1;i>=0;i--){
int j=(x>>i)&1;
if(!nex[p][j]) p=nex[p][!j];
else p=nex[p][j];
}
return ac[p];
}
int getx(int v,int k){
int res=0;
for(int i=k-1;i>=0;i--){
if((v>>i)&1) res=res<<1;
else res=res<<1|1;
}
return res;
}
int main(){
// freopen("input.txt","r",stdin);
int t; cin>>t;
while(t--){
int n; cin>>n>>k;
for(int i=0;i<=n*k;i++){
nex[i][1]=nex[i][0]=0;
}
cnt=0;
for(int i=0;i<n;i++) scanf("%d",&a[i]);
int ansi,ansj,answ=0;
for(int i=0;i<n;i++){
if(i==0){
insert(a[i],i);
continue;
}
int j=find(a[i]);
insert(a[i],i);
int w=(~(a[i]^a[j]))&((1<<k)-1);
if(w>=answ){
answ=w;
ansi=i; ansj=j;
}
}
printf("%d %d %d\n",ansi+1,ansj+1,getx(a[ansi]&a[ansj],k));
}
}