Section 12.Trie树

一、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/AcWing143https://www.acwing.com/problem/content/145/