hdu-统计难题

最新推荐文章于 2021-11-09 00:20:03 发布

原创最新推荐文章于 2021-11-09 00:20:03 发布 · 440 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#字典树 #hdu

瞎折腾的一些东西专栏收录该内容

154 篇文章

订阅专栏

本文介绍了一个使用字典树（Trie）解决单词前缀统计问题的经典案例。通过构建字典树，可以高效地查询以特定字符串为前缀的单词数量，适用于大量单词的快速检索场景。

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

Ignatius最近遇到一个难题,老师交给他很多单词(只有小写字母组成,不会有重复的单词出现),
现在老师要他统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀).

Input
输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是老师交给Ignatius统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串.

注意:本题只有一组测试数据,处理到文件结束.
Output
对于每个提问,给出以该字符串为前缀的单词的数量.
Sample Input

banana
band
bee
absolute
acm

ba
b
band
abc

Sample Output

代码分析:字典树学习的入门(模板)题。

	//hdu-统计难题(字典树)
#include<bits/stdc++.h>
using namespace std;
const int N=(1<<19);
int tcnt;
int n;
int dtrie[N][26]; //i表示树节点编号，26表示节点的儿子编号
int sum[N];//统计查询多少次
void dinsert(char s[],int root)
{
	int l=strlen(s);
	for(int i=0;i<l;i++)
	{
		int cn=s[i]-'a';//计算每个对应数值大小
		
		if(!dtrie[root][cn])
			dtrie[root][cn]=++tcnt; 
		sum[dtrie[root][cn]]++;
		root=dtrie[root][cn];
	}
 } 
int query(char s[]){
	int root=0;
	int l=strlen(s);
	for(int i=0;i<l;i++)
	{
		int cn=s[i]-'a';
		if(!dtrie[root][cn]) 
			return 0;
		root=dtrie[root][cn];
	}
	return sum[root];//返回查询的次数，也就是统计的次数
}
int main(){
	//freopen("1.txt","r",stdin);
	char s[11];
	while(gets(s)!=NULL && s[0]!='\0'){
			dinsert(s,0);
	}
	while(gets(s)){
		printf("%d\n",query(s));
	}
	return 0;
}