hdu 1251 统计难题

最新推荐文章于 2019-08-03 08:20:46 发布

passer__

最新推荐文章于 2019-08-03 08:20:46 发布

阅读量195

点赞数

CC 4.0 BY-SA版权

分类专栏：字典树

本文链接：https://blog.youkuaiyun.com/passer__/article/details/81099976

字典树专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一道使用Trie树解决的HDOJ题目1075，通过实例代码展示了如何构建Trie树进行字符串匹配及计数。包括节点结构定义、插入操作与查询操作。

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

题目链接：

http://acm.hdu.edu.cn/showproblem.php?pid=1075

Trie树的入门题。

入门题懂了基本的这道题就可以过了，就是唯一不一样的是G++超了内存，C++不爆。

代码：

#include<iostream>
#include<string>
using namespace std;
struct node{
	int Count;//统计单词前缀出现次数
	node *next[26];//叶子指针
	int flag;//标记是否构成单词
	node():Count(0),flag(0){//初始化 
		for(int i = 0;i < 26;i++)
			next[i] = NULL;
	} 
};
void insert(node *root,string &s)
{
	node *temp = root;
	int len = s.length();
	for(int i = 0;i < len;i++)
	{
		int id = s[i] - 'a';
		if(temp->next[id] == NULL)
			temp->next[id] = new node();
		temp = temp->next[id];
		temp->Count = temp->Count+1;//标记前缀数量 
	}
	temp->flag = 1;//单词结束 完整单词 
}
int find(node *root,string &s)
{
	node *temp = root;
	int len = s.length(); 
	for(int i = 0;i < len;i++)
	{
		int id = s[i] - 'a';
		if(temp->next[id] != NULL)
			temp = temp->next[id];
		else
			return 0;
	}
	return temp->Count;
} 
int main()
{
	node *root = new node();
	string s;
	int flag=0;
	while(getline(cin,s))
	{
		if(flag)//开始查询 
			printf("%d\n",find(root,s)); 
		else
		{
			if(!s.empty())//非空 
				insert(root,s);
			else//后边输入的为查询 
				flag=1;
		}
	}  
	return 0;
}