给定关键字不在字典中_算法学习笔记(44): 01字典树

01字典树（01-trie）是一种特殊的字典树，它的字符集只有

，主要用来解决一些

异或问题。例如：

（HDU4825 Xor Sum）

Problem Description
Zeus 和 Prometheus 做了一个游戏，Prometheus 给 Zeus 一个集合，集合中包含了N个正整数，随后 Prometheus 将向 Zeus 发起M次询问，每次询问中包含一个正整数 S ，之后 Zeus 需要在集合当中找出一个正整数 K ，使得 K 与 S 的异或结果最大。Prometheus 为了让 Zeus 看到人类的伟大，随即同意 Zeus 可以向人类求助。你能证明人类的智慧么？ Input
输入包含若干组测试数据，每组测试数据包含若干行。
输入的第一行是一个整数T（T < 10），表示共有T组数据。
每组数据的第一行输入两个正整数N，M（<1=N,M<=100000），接下来一行，包含N个正整数，代表 Zeus 的获得的集合，之后M行，每行一个正整数S，代表 Prometheus 询问的正整数。所有正整数均不超过2^32。 Output
对于每组数据，首先需要输出单独一行”Case #?:”，其中问号处应填入当前的数据组数，组数从1开始计算。
对于每个询问，输出一个正整数K，使得K与S异或值最大。

异或、最大，看到这两个词第一反应肯定是线性基，可惜线性基并不能解决这个问题，而需要引入01字典树。

原理很简单，把数字化为二进制后，当作01串，从高位到低位像普通字典树那样存储。例如，如果有数

，存入01字典树就像这样（需要把各个数的位数补成一样多）：

01字典树

现在对于上面问题，我们贪心地解决即可。如果我们要找与给定数异或最大的数，就尽可能走与该数当前位不同的路径。反之则尽可能走与当前位相同的路径。

const int MAXN = 3300000, MAXBIT = 32;
int next[MAXN][2], cnt;
ll num[MAXN];
void init()
{
    memset(next, -1, sizeof(next));
    memset(num, 0, sizeof(num));
    cnt = 0;
}
void insert(ll n)
{
    int cur = 0;
    for (int i = MAXBIT; i >= 0; --i)
    {
        int bit = n >> i & 1; // 求出当前位并插入
        if (next[cur][bit] == -1)
            next[cur][bit] = ++cnt;
        cur = next[cur][bit];
    }
    num[cur] = n;
}
ll find_max(ll x) // 找到与x异或最大的那个数
{
    int cur = 0;
    for (int i = MAXBIT; i >= 0; --i)
    {
        int bit = x >> i & 1;
        if (next[cur][bit ^ 1] != -1) // 优先走与当前位不同的路径
            cur = next[cur][bit ^ 1];
        else
            cur = next[cur][bit];
    }
    return num[cur];
}

（杭电这个题有点毒瘤，卡IO，建议用快读快写。另外上界大于int了，需要开long long。）

再来看一个题：

（洛谷P4551 最长异或路径）

题目描述
给定一棵

个点的带权树，结点下标从

开始到

。寻找树中找两个结点，求最长的异或路径。

异或路径指的是指两个结点之间唯一路径上的所有边权的异或。 输入格式
第一行一个整数

，表示点数。

接下来

行，给出

，分别表示树上的

点和

点有连边，边的权值是

。

输出格式
一行，一个整数表示答案。

这个题还是非常巧妙的，我们知道，一个数异或同一个数两次相当于没有异或。所以

，其实就是

（因为公共段被抵消掉了）。于是我们一趟dfs把每个点到1号点的异或路径求出来，问题就转换成了“

给出一组数，从中选两个数异或，求最大值”。也就是对每一个数分别解决“ 从一组数中选一个数与给定数异或最大”的子问题。

一般来说，一组数中取任意个数求最大/最小异或和可以用线性基，取两个数求最大/最小异或和则可以用01字典树解决。

---

Pecco：算法学习笔记（目录）​zhuanlan.zhihu.com