hdu1251 统计难题(Trie树入门题)

最新推荐文章于 2021-08-23 17:24:57 发布

原创最新推荐文章于 2021-08-23 17:24:57 发布 · 1.1k 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

hdu

当前文章被收录于：

hdu

83 篇文章 0 人学习

订阅专栏查看详情

字符串--------------

当前文章被收录于：

字符串--------------

22 篇文章 0 人学习

订阅专栏查看详情

字典树

当前文章被收录于：

字典树

1 篇文章 0 人学习

订阅专栏查看详情

当前文章被以下社区和专栏收录：

本文介绍了一道使用Trie树解决的经典ACM竞赛题目，并提供了详细的C语言实现代码。通过该示例，读者可以了解如何构建Trie树、插入字符串以及搜索特定字符串出现的次数。

题目链接：

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

Trie树的入门题。

代码：

#include<stdio.h>
#include<string.h>
const int maxnode=1000005;
const int sigma_size=26;
struct Trie{
	int ch[maxnode][sigma_size];
	int sz;
	int num[maxnode];
	Trie(){ 
		sz=1;
	 	memset(ch[0],0,sizeof(ch[0]));
	 	memset(num,0,sizeof(num));
	 }
	int idx(char c)  { return c - 'a'; }
	
	
	void insert(char *s, int v){
		int  u=0, n =strlen(s);
		for(int i=0;i<n;i++){
			int c= idx(s[i]);
			if(!ch[u][c]){
				memset(ch[sz],0,sizeof(ch[sz]));
				
				num[sz]=0;
				ch[u][c]=sz++;
			}
			u=ch[u][c];
			num[u]++;
		} 
	}


	int search(char *s){
		int n=strlen(s);
		int u=0;
		for(int i=0;i<n;i++){
			int c=idx(s[i]);
			if(!ch[u][c])
				return 0;
			
			u=ch[u][c];
			
		}
		return num[u];
	}
};
Trie T;
int main()
{
	char s[15];
	int flag=0;
	
	while(gets(s)){
		if(s[0]=='\0')break;
		T.insert(s,1);
	}
	
	while(~scanf("%s",s)){
		printf("%d\n",T.search(s));
	}
	
}