Trie树

本篇小博客主要介绍Trie树得基本知识，以及两道例题 来感受一下代码的实现。

基本概述

1. 用来高效存储和查找字符串集合的数据结构
2. 比如我们想在trie树存储下面的单词：
   【从根节点开始往后遍历每一个字符，若无当前字母就创建结点】

标记：在当前点打一个标记，表示以该结点结尾的，是有一个单词的。如图中的五角星。

3. 查找：
   查找一个存在的单词：从根节点开始走，依次往后遍历，看到该单词的最后一个字母打着标记。
   查找一个不存在的单词：从根节点开始发现单词中的某一个字母未出现在合法路径中，或者整个路径上得点都在，但最后一个字母未标记。

Trie字符串统计

题目描述：

题目链接

几点说明：

1. 本题说只含有小写字母，因此每个结点最多向外连26条边。

2. son数组便是树的存储了，cnt表示以当前这个点结尾的单词有多少个。idx和单链表中的一样表示我们用到的下标，idx=0 既是根节点又是空节点

#include <iostream>
#include <algorithm>
using namespace std;
const int N = 100010;
int son[N][26], cnt[N], idx;
char str[N];
void insertt(char str[]){ //c插入操作
    int p = 0;
    for(int i = 0; str[i]; i ++){
        int u = str[i] - 'a';
        if(!son[p][u]) son[p][u] = ++idx;
        p = son[p][u];
    }
    cnt[p] ++; //p最后停在最后一个字母上。
}
//查询函数
int query(char str[]){
    int p = 0;
    for(int i = 0; str[i]; i ++){
        int u = str[i] - 'a';
        if(!son[p][u]) return 0;
        p = son[p][u];
    }
    return cnt[p];
}

int main(void){
    int n;
    cin >> n;
    while(n --){
        char op[2];
        cin >> op >> str;
        if(op[0] == 'I') insertt(str);
        else cout << query(str) << endl;
    }
    return 0;
}

最大异或对

题目描述：

题目链接

思路：

1. 我们不妨看一下暴力代码，然后再想着去优化。
   暴力代码必超时：
   
   我们现在就是要优化。
   暴力的想法：
   枚举Ai, 从[A1, An]中选出一个值Aj, 使得Ai ^ Aj最大，再从A1 -Ai中选出最大值中的最大值。
   枚举Ai不用优化，关键是找Aj.
   假设Ai的二进制表示 = 11001100101011…
   从最高位依次往后寻找这二进制位是否有值为目标的二进制值，使得两数在该位的异或值为1.若无则进行下一位的寻找，若有，则从这些符合该位的数中继续寻找。这样寻找来的数是异或值最大的。
   可参考下图：
   
   顺时针旋转90度，便可得到trie树。

2. 把所有数建立一个tire树。
   
   
   Trie树的结点个数：M = 30 * 1e5 = 3e6

#include <iostream>
#include <algorithm>
using namespace std;
const int N = 1e5 + 10, M = 3e6 + 10;
int son[M][2], idx, arr[N];
int n;

void insertt(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 query(int x){
    int res = 0, p = 0;
    for(int i = 30; i >= 0; i --){
        int s = x >> i & 1;
        if(son[p][!s]){
            res += 1 << i;
            p = son[p][!s];
        }
        else{
            p = son[p][s];
        }
    }
    return res;
}

int main(void){
    cin >> n;
    for(int i = 0; i < n; i ++){
        cin >> arr[i];
        insertt(arr[i]);
    }
    int res = 0;
    for(int i = 0; i < n; i ++) res = max(res, query(arr[i]));
    cout << res;
    return 0;
}