hdu 1251 统计难题

最新推荐文章于 2019-08-09 09:05:28 发布

希望坚持

最新推荐文章于 2019-08-09 09:05:28 发布

阅读量575

点赞数

分类专栏：字典树文章标签：字典树 hdu 简单题 acm

本文链接：https://blog.youkuaiyun.com/u013022194/article/details/38470497

版权

字典树专栏收录该内容

1 篇文章

订阅专栏

题目链接：点我点我~

统计难题

Time Limit: 4000/2000 MS (Java/Others) Memory Limit: 131070/65535 K (Java/Others)
Total Submission(s): 18175 Accepted Submission(s): 7957

Problem Description

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

Input

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

注意:本题只有一组测试数据,处理到文件结束.

Output

对于每个提问,给出以该字符串为前缀的单词的数量.

Sample Input

      
      
       
       banana
band
bee
absolute
acm

ba
b
band
abc

Sample Output

题意：找出同样前缀的字符串个数，输出；

该题是一道标准的用字典树的题目，字典树就是由字符组成的一棵树.

只要给每个节点赋上支路数，就可以表示同样的前缀字符串个数了。

套了模板。。。。。

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

struct node 
{
	struct node *br[26];
	int num;
};

node *root;

void trie_init()//字典树清空
{
	root=new node;
	root->num=0;
	for(int i=0;i<26;i++)
	{
		root->br[i]=NULL;
	}
}

void trie_insert(char str[])//加入字典树
{
	int len=strlen(str);
	node *s=root;
	for(int i=0;i<len;i++)
	{
		int id=str[i]-'a';
		if(s->br[id]==NULL)
		{
			node *t=new node;
			for(int j=0;j<26;j++)
			{
				t->br[j]=NULL;
			}
			t->num=0;
			s->br[id]=t; 
		}
		s=s->br[id];
		s->num++;
	}
}

int trie_find(char str[])//在字典树中查找
{
	int len=strlen(str);
	node *s;
	s=root;
	int count=0;
	for(int i=0;i<len;i++)
	{
		int id=str[i]-'a';
		if(s->br[id]==NULL)
			return 0;
		else
		{
			s=s->br[id];
			count=s->num;
		}
	}
	return count;
}

int main()
{
	char str[15];
	trie_init();
	while(gets(str),strcmp(str,""))
	{
		trie_insert(str);
	}
	while(gets(str))
	{
		int res=trie_find(str);
		printf("%d\n",res);
	}
	return 0;
}