HDOJ 1251 统计难题(j简单的字典树trie的应用)

原创于 2015-02-11 15:57:07 发布 · 500 阅读

0 ·

CC 4.0 BY-SA版权

字符串处理专栏收录该内容

10 篇文章

订阅专栏

本文介绍了一种字典树（Trie）的数据结构实现方式，并提供了完整的C++代码示例。通过该实现可以高效地进行字符串查找、插入等操作。文章详细解释了如何构建字典树节点、如何插入新单词以及如何查找特定单词出现的次数。

整理一下自己的字典树模板。

#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <algorithm>
#include <vector>
using namespace std;
const int MAXN = 15;
typedef struct Trie_Node
{
        struct Trie_Node *next[26];
	int val;
}Trie;
void insert(Trie *root, char *s)
{
	Trie *p = root;
	int i = 0;
	while(s[i] != '\0')
	{
		if(p->next[s[i]-'a'] == NULL)
		{
			Trie *temp = new Trie;
			for(int j=0;j<26;j++)
				temp->next[j] = NULL;
			temp->val = 0;
			p->next[s[i]-'a'] = temp;
		}
		p = p->next[s[i]-'a'];
		p->val++;
		i++;
	}
}
int Find(Trie *root, char *s)
{
	Trie *p = root;
	int i = 0;
	while(s[i] != '\0')
	{
		if(p->next[s[i]-'a'] == NULL)
			return 0;
		p = p->next[s[i]-'a'];
		i++;
	}
	return p->val;
}
int main()
{
	char s[20];
	Trie *root = new Trie;
	for(int i=0;i<26;i++)
		root->next[i] = NULL;
	root->val = 0;
	while(gets(s) && strlen(s)) insert(root, s);
	while(gets(s))
	{
		printf("%d\n", Find(root,s));
	}
	return 0;
}