acautomaton
acautomaton
发布于 2022-02-10 / 11 阅读
0
0

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];
}

三、配套练习

AcWing835

AcWing143


评论