01Trie树 CF923C Perfect Security

Posted 2021-01-17 wangxiaodai

CF923C Perfect Security

上下各n个数,求一种排列p,使上面的数i异或pi成为新的数i,求方案另字典序最小,输出该结果

01Trie树。

记录每个节点经过多少次。

每一次查询的时候将经过的点的标记加1.

那么当我们要去走某一个节点的时候，就去判断他的标记是否小于初始的值，如果不是那么就不能走。

教训告诉我们一定要想好细节再打代码，我没想好怎么处理初始标记就去写了个拓扑排序。低智低智。。。

code：

#include <iostream>
#include <cstdio>
#include <queue>
#include <cstring>

using namespace std;

const int wx=300017;

inline int read(){
    int sum=0,f=1; char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1; ch=getchar();}
    while(ch>='0'&&ch<='9'){sum=(sum<<1)+(sum<<3)+ch-'0'; ch=getchar();}
    return sum*f;
}

queue<int > q;
int a[wx];
int p[wx];
int n;

struct Trie{
    int tr[wx*33][2];
    int cnt;
    int son[wx*33];
    int tot[wx*33];
    
    void clear(){
        memset(son,0,sizeof son);
        memset(tot,0,sizeof tot);
    }
    
    void insert(int x){
        int p=0;
        for(int i=33;i>=0;i--){
            int k=((x&(1ll<<i))?1:0);
            if(!tr[p][k])tr[p][k]=++cnt;
            p=tr[p][k]; son[p]++;
        }
        q.push(p);
    }
    
    int query(int x){
        int p=0; int re=0;
        for(int i=33;i>=0;i--){
            int k=((x&(1ll<<i))?1:0);
            if(tr[p][k]&&tot[tr[p][k]]<son[tr[p][k]])
                p=tr[p][k];
            else p=tr[p][k^1],re+=(1ll<<i);
            tot[p]++;
        }
        return re;
    }
    
}Trie;

int main(){
    n=read(); Trie.clear();
    for(int i=1;i<=n;i++)a[i]=read();
    for(int i=1;i<=n;i++)p[i]=read(),Trie.insert(p[i]);
    for(int i=1;i<=n;i++)printf("%d ",Trie.query(a[i]));
    puts("");
    return 0;
}