ACM模板——字典树

01字典树

/*
数组大小定义为：2e6+10，数组大小与 n，m 是无关的，只与 cid 的最大值相关，算出对应的最多位数，比如：此题目中 cid 最大值为 1000000，对应的二进制位数最多 20位，根据完全二叉树的结点公式，最多 2^21-1。因为这里是边作为 二进制位，结点作为 链接下一个结点的下标。又因为题目说了是 cid^pid 最大的数，所以根据异或相同则0，不同则1的性质可知，只要是高位二进制各异就是最大的数了，所以 query 函数先尽量找与自己不一样的二进制，实在找不到的话，再选择自己的二进制位。
*/

#include<bits/stdc++.h>
#include<cmath>

#define mem(a,b) memset(a,b,sizeof a);
#define INF 0x3f3f3f3f

using namespace std;

typedef long long ll;
const int maxn=1e5+10;

int n,m,cnt;
int trie[21*maxn][2];
int val[21*maxn];

void init()
{
    cnt=0;
    mem(val,0); mem(trie,0);
}

void add(int cid)
{
    int rt=0,id;
    for(int i=19;i>=0;i--) // 从 19 开始是因为从最高位开始操作
    {
        id=(cid>>i)&1;
        if(!trie[rt][id]) trie[rt][id]=++cnt; // trie[rt][id] 不能等于 0，以免与初始化混淆
        rt=trie[rt][id];
    }
    val[rt]=cid;
}

int query(int pid)
{
    int rt=0,id;
    for(int i=19;i>=0;i--) // 同上
    {
        id=(pid>>i)&1;
        if(trie[rt][id^1]) rt=trie[rt][id^1];
        else rt=trie[rt][id];
    }
    return val[rt];
}