一、Trie树的基本作用:
- 高效地存储和查找字符串集合的数据结构
例如,我们要存储以下字符串:abcdef,abdef,aced,bcdf,bcff,cdaa,abc,我们可以以树的形式存储:
在这里,我们以标红的形式来标记每个字符串的末尾。
二、代码实现
初始化
#include <iostream>
using namespace std;
const int N = 100010;
char str[N];
int son[N][26]; //所有节点
int cnt[N]; //以当前节点结尾的单词个数
int idx = 0; //当前用到了哪个下标
插入操作
void insert(char str[])
{
int p = 0; //当前节点
for (int i = 0; str[i] != '\0'; i++)
{
int u = str[i] - 'a'; //当前字母对应的子节点编号
if (!son[p][u]) //如果这个节点不存在,就把它创建出来
son[p][u] = ++idx;
p = son[p][u]; //移动到下一个节点
}
cnt[p]++; //以这个点为结尾的单词数
}
查询操作
int query(char str[])
{
int p = 0;
for (int i = 0; str[i] != '\0'; i++)
{
int u = str[i] - 'a';
if (!son[p][u])
return 0;
p = son[p][u];
}
return cnt[p];
}
三、配套练习
AcWing835https://www.acwing.com/problem/content/837/AcWing143
https://www.acwing.com/problem/content/145/