codeforces补题

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) aix)&(ajx)
最大。
思路
要使得这个式子最大我们首先得选出最优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));
    }
}   

你可能感兴趣的:(数据结构和算法,算法,数据结构)