[CodeForces948D]Perfect Security（01字典树）

Description

题目链接

Solution

01字典树模板题，删除操作用个数组记录下就行了

Code

#include <cstdio>
#include <algorithm>
#include <cmath>

int n,T[9000010][2],v[9000010],A[300010],num[9000010],rt=1,B[300010];

inline 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 Insert(int x,int id){
	int u=0,op;
	for(int i=30;i>=0;--i){
		op=((x&(1<<i))!=0);
		if(!T[u][op]) T[u][op]=rt++;
		u=T[u][op];
		num[u]++;
	}
	v[u]=id;
}

void del(int x){
	int u=0,op;
	for(int i=30;i>=0;--i){
		op=((x&(1<<i))!=0);
		u=T[u][op];
		num[u]--;
	}
}

int get(int x){	
	int u=0,op;
	for(int i=30;i>=0;--i){
		op=((x&(1<<i))!=0);
		if(T[u][op]&&num[T[u][op]]) u=T[u][op];
		else u=T[u][op^1];
	}
	del(A[v[u]]);
	return x^A[v[u]];
}

int main(){
	n=read();
	for(int i=1;i<=n;++i) B[i]=read();
	for(int i=1;i<=n;++i) Insert(A[i]=read(),i); 
	for(int i=1;i<=n;++i) printf("%d ",get(B[i]));
	return 0;
}