python

4.5 KiB

Raw Blame History Unescape Escape

This file contains ambiguous Unicode characters that may be confused with others in your current locale. If your use case is intentional and legitimate, you can safely ignore this warning. Use the Escape button to highlight these characters.

##AcWing 143. 最大异或对

一、题目描述

在给定的 N 个整数 A_1，A_2……A_N 中选出两个进行 xor（异或）运算，得到的结果最大是多少？

输入格式 第一行输入一个整数 N。

第二行输入 N 个整数 A_1～A_N。

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

数据范围 1≤N≤10^5,0≤Ai<2^{31}

输入样例：

3
1 2 3

输出样例：

二、分析思路

先来思考暴力怎么做:

// 最大异或对,用暴力是超时的
// 通过了 6/10个数据
#include<iostream>
using namespace std;
const int N=1e5+10;
int a[N];
int res;
int main(){
    int n;
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i];
    
    for(int i=1;i<=n;i++)
        for(int j=i+1;j<=n;j++)
            res=max(res,a[i]^a[j]);
    cout<<res<<endl;
    return 0;
}

结果不出意外，TLE,只能想办法进行优化

`Trie`树思路

1、将整数解析为二进制数，即有符号整数，31位，就是0-30，按Trie树进行存储, 整数的Trie树存储。

2、每个数字的每一个二进制位，需要从高位到低位，即for(int i = 30; i >= 0; i--)，想像一下你在构建一个Trie树，那么根root就是最高位，然后一路走到31位，就是最低位。

3、每个数字想要找到与自己形成最大异或值的另一个数字，我们现在已经把它们保存到Trie树里了，那怎么找呢？什么样的两个数字才是最大异或值的对呢？就是每一位完全相反的就肯定是最大的异或对！那如果某一位相反的结点并不存在呢？这就是退而求其次的思路了，我们尽量从左到右找出与当前数字本位相反的路径，如果存在，就继续探索，如果不存在，那就使用一样的本位值。这样下来，到31位，就可以找到和自己匹配最大的异或值。

总结一下

Trie里可以用来保存数字，数字需要通过二进制（由高位到低位）进行保存。
增加一个数字进来，其实就是增加了一个层级为31级的 模拟字符串
放入一个数字，那么它肯定会在任意一级（共31级）存在一边，另一边可能存在，也可能不存在。

三、实现代码

#include <bits/stdc++.h>

using namespace std;

const int N = 1e5 + 10;
const int M = N * 31;

int n, res;
int a[N];
int tr[M][2];
int idx;

// 构建数字二进制位的Trie树
void insert(int x) {
    int p = 0;
    for (int i = 30; i >= 0; i--) {
        int u = (x >> i) & 1;            // 取出当前位的值
        if (!tr[p][u]) tr[p][u] = ++idx; // 构建Trie树
        p = tr[p][u];
    }
}

// 所谓与x异或最大，就是利求在高位上尽量不一样，如果找不到不一样的，就只能找一样的，下一个继续优先找不一样的
// 在Trie树中查找到与x异或最大的数
int query(int x) {
    int p = 0, ans = 0;
    for (int i = 30; i >= 0; i--) {
        int u = (x >> i) & 1; // 取出x的当前二进制位
        if (tr[p][!u]) {      // 如果存在可以异或的路可以走的话,尽量先走
            p = tr[p][!u];
            ans = ans * 2 + !u; // 还原二进制数字为十进制
        } else {
            p = tr[p][u];      // 否则只能走与自己本位一样的路线
            ans = ans * 2 + u; // 还原二进制数字为十进制
        }
    }
    return ans;
}

int main() {
    cin >> n;
    for (int i = 1; i <= n; i++) cin >> a[i], insert(a[i]);

    for (int i = 1; i <= n; i++) {
        int t = query(a[i]);
        res = max(res, a[i] ^ t);
    }
    printf("%d", res);
    return 0;
}

五、对于一维数据范围的思考

无论是模板题还是最大异或对着一题都有这么一行代码 if (!tr[p][u]) tr[p][u] = ++ idx; p = tr[p][u];

所以我们可以知道，tr数组的一维下标最大值的选取实际上是跟idx能够自增多少次来决定的

AcWing 835. Trie字符串统计 中，输入的字符串总长度不超过 10^5，所以一维值选取1e5+10

而在 AcWing 143. 最大异或对 中，数字需要以2进制进行表示，而每个数字最大为2的31次幂,所以一维下标应为数字的个数*31。

4.5 KiB Raw Blame History Unescape Escape