Acwing基础算法Trie字典树之最大异或对

最大异或对

143. 最大异或对 - AcWing题库

需求：找到最大的异或对

原理：将数转为二进制存储为tire树。

问题一：转二进制

//获取二进制表示的x的第i位
x >> i & 1;

问题二：求最大异或

如果当前位为1，则找为0的节点。如果当前位没有相异，则使用当前位数值。从最高位开始查找，保证最大。

int p = 0;
for(int i = 30; i >= 0; i --)
{
    int u = x >> i & 1;
    //相异位存在
    if(q[p][!u]) p = q[p][!u];
    else p = q[p][u];
}

问题三：查找过程将二进制转为十进制

//十进制转二进制(t = 1 or 0)
int t = ans%2;
ans/=2;
//倒过来就是二进制转为十进制(u = 1 or 0)
ans = ans * 2 + u;

问题四：为什么可以一边插一边比较

异或前后不分，因此会有重复。排列组合为
$$C_n^2=n(n-1)/2$$

所以只需要一边插入一边算就行了，之后的会将已经插入的也算进去。

#include<iostream>
using namespace std;

//最大个数和最大位数
const int N = 100010, M = N * 31;
int q[M][2],idx;

void insert(int x)
{
    int p = 0;
    for(int i = 30; i>= 0; i --)
    {
        int u = x >> i & 1;
        if(!q[p][u]) q[p][u] = ++ idx;
        p = q[p][u];
    }
}

int query(int x)
{
    int p = 0;
    int ans = 0;
    for(int i = 30; i >=0; i --)
    {
        int u = x >> i & 1;
        if(q[p][!u])
        {
            p = q[p][!u];
            ans = ans *2 + !u;
        }
        else
        {
            p = q[p][u];
            ans = ans *2 +u;
        }
    }
    
    return ans;
}

int main()
{
    int n;
    cin >> n;
    int res = 0;
    for(int i = 0; i < n; i ++ )
    {
        int a;
        cin >> a;
        insert(a);
        int t = query(a);
        res = max(res,a ^ t);
    }
    
    cout << res;
    
}