【Trie】AcWing143. 最大异或对

AcWing 143. 最大异或对

题解

将每个数的二进制形式存入Trie树，再对每个数的每一位进行遍历，若为1则找0，若为0则找1。

#include <iostream>

using namespace std;

const int N = 1e5 + 10, M =  3100010;  //每个数保存31位，故1e5*31

int son[M][2], a[N], idx;


//因为每个数都是31位，故我们一定知道结尾不用标记结尾
void insert(int x)
{
    int p = 0;
    for(int i = 30; i >= 0; --i)
    {
        int &s = son[p][x >> i & 1];
        if(!s) s = ++ idx;
        p = s;
    }
}

int search(int x)
{
    int p = 0;
    int res = 0;
    for(int i = 30; i >= 0; --i)
    {
        int s = x >> i & 1;
        if(son[p][!s])
        {
            p = son[p][!s];
            res +=  1 << i;
        }
        else p = son[p][s];
    }
    return res;
}

int main()
{
    int n;
    cin >> n;
    for(int i = 0; i < n; ++i) cin >> a[i], insert(a[i]);
    int ans = 0;
    for(int i = 0; i < n; ++i) ans = max(ans, search(a[i]));
    cout << ans << endl;
    return 0;
}