HDU1251_统计难题_字典树（tire）查前缀含词条数

统计难题

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

Problem Description

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

 

Input

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

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

 

Output

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

 

Sample Input

  
  

   
   banana
band
bee
absolute
acm

ba
b
band
abc
  
  

 

Sample Output

  
  

   
   2
3
1
0
  
  

#include<cstdio>
#include<cstring>

typedef struct Tire_node
{
	int cnt;
	Tire_node* next[26];

	Tire_node(){
	cnt=0;
	memset(next,0,sizeof(next));
	}

}tire,*tirep;

char str [15];

tirep Creat_tire ()
{
	tirep x=new(tire);
	return x;
}

int main ()
{
	//freopen("in.txt","r",stdin);

	tire T;

	while(gets(str)&&*str){

		int len=strlen(str);
		tirep p=&T;

		for(int i=0;i<len;i++){

			if(p->next[str[i]-97]==NULL)
				p->next[str[i]-97]=Creat_tire();

			p=p->next[str[i]-97],p->cnt++;
		}

	}

	while(gets(str)){

		int len=strlen(str);
		tirep p=&T;

		int i=0;
		for(;i<len;i++)
			if(p->next[str[i]-97]==NULL) break;
			else p=p->next[str[i]-97];

		if(i>=len) printf("%d\n",p->cnt);
		else printf("0\n");

	}

	return 0;

}