【trie树】the xor largest pair

LOJ #10050. The XOR Largest Pair

trie树三连击biu~

《进阶指南》P72 例题

思路与上题无异， 只细节不同qaq

1. 分解二进制，高位在前， 建立trie树（0/1）
2. search操作：首选不同的（xor值才能大）， 就等于在后面加了一个1， 所以 sum = (sum<<1)|1；
3. 如果没有不同的， 就等于在后面加了一个0， 所以 sum = (sum<<1);
4. (唔为什么变紫色了啊啊啊
5. 如果从32开始枚举， 爆int 记得开long long ， （gjh：多枚举几位， 开个long long保险啊！ ）这题从31开始也可以， 只要int即可
6. 后面search(a[i]) 写成search(i) 可还行QAQ

代码上！

 

#include<cstdio>
#include<iostream>
using namespace std;
const int sz = 100010;
int n, tot = 0;
int trie[sz*32][2], a[sz];
void insert(int x) {
    int root = 0;
    for(int i = 31; i >= 0; i--) {
        int id = (x>>i)&1;
        if(!trie[root][id]) trie[root][id] = ++tot;
        root = trie[root][id];
    }
}
int search(int x) {
    int root = 0, sum = 0;
    for(int i = 31; i >= 0; i--) {
        int id = (x>>i)&1;
        if(trie[root][id^1]) {
            root = trie[root][id^1];
            sum = (sum<<1)|1;
        }
        else {
            root = trie[root][id];
            sum = sum<<1;
        }
    }
    return sum;
}
int main() {
    int ans = 0;
    scanf("%d", &n);
    for(int i = 1; i <= n; i++) {
        scanf("%d", &a[i]);
        insert(a[i]);
    }
    for(int i = 1; i <= n; i++) {
        if(search(a[i])>ans) {
            ans = search(a[i]);
        }
    }
    printf("%d\n", ans);
    return 0;
}