本篇小博客主要介绍Trie树得基本知识,以及两道例题 来感受一下代码的实现。
基本概述
- 用来高效存储和查找字符串集合的数据结构
- 比如我们想在trie树
存储
下面的单词:
【从根节点开始往后遍历每一个字符,若无当前字母就创建结点】
标记:在当前点打一个标记,表示以该结点结尾的,是有一个单词的。如图中的五角星。
查找:
查找一个存在的单词:从根节点开始走,依次往后遍历,看到该单词的最后一个字母打着标记。
查找一个不存在的单词:从根节点开始发现单词中的某一个字母未出现在合法路径中,或者整个路径上得点都在,但最后一个字母未标记。
Trie字符串统计
题目描述:
题目链接
几点说明:
-
本题说只含有小写字母,因此每个结点最多向外连26条边。
-
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;
}
最大异或对
题目描述:
思路:
-
我们不妨看一下暴力代码,然后再想着去优化。
暴力代码必超时:
我们现在就是要优化
。
暴力的想法:
枚举Ai, 从[A1, An]中选出一个值Aj, 使得Ai ^ Aj最大,再从A1 -Ai中选出最大值中的最大值。
枚举Ai不用优化,关键是找Aj.
假设Ai的二进制表示 = 11001100101011…
从最高位依次往后寻找这二进制位是否有值为目标的二进制值,使得两数在该位的异或值为1.若无则进行下一位的寻找,若有,则从这些符合该位的数中继续寻找。这样寻找来的数是异或值最大的。
可参考下图:
顺时针旋转90度,便可得到trie树。 -
把所有数建立一个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;
}