统计难题

统计难题

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

Problem Description

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

 

Input

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

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

 

Output

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

 

Sample Input

banana band bee absolute acm ba b band abc

 

Sample Output

2 3 1 0

题解：字典树问题

 AC code:

#include <iostream>
#include <string>
using namespace std;
struct T 
{
	int num;
	T* next[26];
	T()
	{
	   num = 0;
	   for(int i=0; i<26; i++)
	   {
		   next[i] = NULL;
	   }
	}
}t;
void insert(char s[])
{
   T* p = &t;
   for(int i=0; s[i]; i++)
   {
	   int a = s[i]-'a';
	   if(p->next[a] == NULL)
		   p->next[a] = new T();
	   p = p->next[a];
	   p->num++;
   }
}
int find(char s[])
{
	T* p = &t;
	for(int i=0; s[i]; i++)
	{
       int a = s[i]-'a';
	   if(p->next[a]==NULL)
		   return 0;
	   p = p->next[a];
	}
   return p->num;
}
int main()
{
	char s[11];
   while(gets(s)&&strcmp(s,"")!=0)
   {
	   insert(s);
   }
   while(scanf("%s",&s)!=EOF)
	   cout<<find(s)<<endl;

 return 0;
}