923c C. Perfect Security

Posted 2020-10-28 invoid

Trie树。

要求字典序最小，所以由前到后贪心的选择。建一个trie树维护b数列。

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int maxn = 300000*(20+5) ;
const int maxm = 30 + 10;

int a[maxn],s[maxn],p0[maxn],p1[maxn],cnt;
int n,b;

int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)f=-1;ch=getchar();}
    while(ch>=‘0‘&&ch<=‘9‘){x=x*10+ch-‘0‘;ch=getchar();}
    return x*f;
}

void build() {
    n=read();
    for(int i=1;i<=n;i++) a[i]=read();
    for(int i=1,rhs,t;i<=n;i++) {
        rhs=read();
        t=0;
        for(int j=29;j>=0;j--) 
        {
            b=(rhs>>j&1);
            if(b==0) 
            {
                if(!p0[t]) p0[t]=++cnt; 
                t=p0[t];
            }
            else 
            {
                if(!p1[t]) p1[t]=++cnt; 
                t=p1[t];
            }
            s[t]++;
        }    
    }
    
    for(int i=1,k,t;i<=n;i++) 
    {
        k=t=0;
        for(int j=29;j>=0;j--) 
        {
            b=(a[i]>>j)&1;
            if(b==0)
                if(p0[t]&&s[p0[t]]) 
                {
                    t=p0[t];
                    k*=2;
                    //printf("0");
                }
                else
                {
                    t=p1[t];
                    k=k*2+1;
                    //printf("1");    
                }
            else 
            {
                if(p1[t]&&s[p1[t]]) 
                {
                    t=p1[t];
                    k=k*2+1;    
                }
                else 
                {
                    t=p0[t];
                    k*=2;
                }
            }
            s[t]--;
        }
        printf("%d ",a[i]^k);
    }
    printf("\n");
}

void solve() 
{
    
}

int main() 
{
    build();
    solve();
    
    return 0;
}