＜模板＞Trie字典树

最新推荐文章于 2024-10-07 09:58:35 发布

默默可书虫

最新推荐文章于 2024-10-07 09:58:35 发布

阅读量66

点赞数

分类专栏：算法模板文章标签：字符串算法数据结构

本文链接：https://blog.csdn.net/a1121420/article/details/116158854

版权

算法模板专栏收录该内容

12 篇文章 0 订阅

订阅专栏

该博客介绍了如何使用字典树（Trie）数据结构来高效地插入和搜索字符串。通过示例代码展示了如何构建字典树并进行插入操作，同时提供了查找字符串是否存在于字典树中的方法。在主函数中，演示了向字典树插入多个字符串并进行搜索的流程。

摘要由CSDN通过智能技术生成

字典树，一种存储字符串的结构，可以快速的找到某一字符串是否在字典树中。

int trie[SIZE][26],tot=1;  //假设输入为26个小写字母

void insert(char* str)   //插入数据
{
	int len=strlen(str),p=1;
	for(int k=0;k<len;k++)
	{
		int ch=str[k]-'a';
		if(trie[p][ch]==0) trie[p][ch]=++tot;
		p=trie[p][ch];
	}
	endk[p]=true;
}

bool search(char* str)   //查找是否在字典树中
{
	int len=strlen(str),p=1;
	for(int k=0;k<len;k++)
	{
		if(trie[p][str[k]-'a']==0) return false;
		p=trie[p][str[k]-'a'];
	}
	return endk[p];
}

具体的应用例子：

#include<iostream>
#include<cstring>
using namespace std;

const int SIZE = 100;
int trie[SIZE][26], tot = 1;
bool endk[SIZE];
//插入字典树
void insert(char* str)
{
	int len = strlen(str), p = 1;
	for (int k = 0; k < len; k++)
	{
		int ch = str[k] - 'a';
		if (trie[p][ch] == 0)trie[p][ch] = ++tot;
		p = trie[p][ch];
	}
	endk[p] = true;

}

//查找是否在字典树
bool search(char* str)
{
	int len = strlen(str), p = 1;
	for (int k = 0; k < len; k++)
	{
		p = trie[p][str[k] - 'a'];
		if (p == 0) return false;
	}
	return endk[p];
}
int main()
{
	char s[10];
	for (int i = 0; i < 3; i++)
	{
		cin >> s;
		insert(s);
	}
	cin >> s;
	cout << search(s);
	return 0;
}