Codeforces 923 C. Perfect Security

http://codeforces.com/contest/923/problem/C

Trie树

#include<cstdio>
#include<iostream>

using namespace std;

#define N 300001
#define M 30

int tot=,tr[N*M][],sum[N*M];

int a[N];

void read(int &x)
{
    x=; char c=getchar();
    while(!isdigit(c)) c=getchar();
    while(isdigit(c)) { x=x*+c-''; c=getchar(); }
}

void insert(int x)
{
    int now=;
    sum[now]++;
    for(int i=M-;i>=;--i)
    {
        if(!tr[now][x>>i&]) tr[now][x>>i&]=++tot;
        now=tr[now][x>>i&];
        sum[now]++;
    }
}

int find(int x)
{
    int now=,y=;
    sum[now]--;
    for(int i=M-;i>=;--i)
    {
        if(sum[tr[now][x>>i&]]) now=tr[now][x>>i&];
        else now=tr[now][!(x>>i&)],y+=<<i;
        sum[now]--;
    }
    return y;
}

int main()
{
    int n,x;
    read(n);
    for(int i=;i<=n;++i) read(a[i]);
    for(int i=;i<=n;++i)
    {
        read(x);
        insert(x);
    }
    for(int i=;i<=n;++i) printf("%d ",find(a[i]));
    return ;
}