trie树

最新推荐文章于 2025-03-09 10:32:18 发布

chen<>

最新推荐文章于 2025-03-09 10:32:18 发布

阅读量278

点赞数

分类专栏：基础算法文章标签：算法数据结构 c++

本文链接：https://blog.youkuaiyun.com/weixin_53051813/article/details/127132944

版权

基础算法专栏收录该内容

17 篇文章

订阅专栏

本文介绍了Trie树数据结构的原理与应用，并通过具体的示例进行解释。文中还提供了详细的C++代码实现，包括如何构建Trie树、插入字符串以及查询特定字符串出现的次数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

今天来学习一种数据结构：trie树，它可以很高效的存储和查找字符串

1、示例

假设现在需要存储3个字符串，分别是abc,ade,abd,那么将三个字符串存储到trie树中什么样子呢？
在这里插入图片描述

2、代码

儿子数组son[p][i]:存储从节点p沿着i这条边走到的子节点。边为26个小写字母对应的映射值0~25，每个节点最多可以有26个分叉。
计数数组cnt[p]:存储以节点p结尾的单词的插入次数
节点编号：用来给节点编号
建字典树：

空trie仅有一个根节点，编号为0
从跟开始插，枚举字符串的每个字符，如果有儿子，则p指针走到儿子。如果没有儿子，则先创建儿子,p指针再走到儿子。
在单词结束点记录插入次数

// 时间：2021.12.23 23点06分
// 数据结构：trie树 
#include<iostream>
using namespace std;
 
const int N=100010;
char str[N];
int son[N][26],cnt[N],idx;
//son[x]的所有儿子都存在son[x]里 
//cnt[X]存的是以x结尾的单词的个数 
//idx表示正在使用的节点  
//下标是0的点，既是根节点，又是空节点

void insert(char str[]) 
{
	int p=0;
	for(int i=0;str[i];i++)
	{
		int u=str[i]-'a';
		if(!son[p][u]) son[p][u]=++idx;//如果p节点没有u儿子，则创建一个
		p=son[p][u];//循环完成以后，p就在最后一个位置 
	 } 
	 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 ()
 {
 	int n;
 	scanf("%d",&n);
 	while(n--)
 	{
 		char op[2];
 		scanf("%s%s",op,str);
 		if(op[0]=='I') insert(str);//插入操作
 		else printf("%d\n",query(str));//查询操作
	 }
	 return 0;
 }