HDOJ 1251 统计难题（字典树）



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

【科普】什么是BestCoder？如何参加？

统计难题 Time Limit: 4000/2000 MS (Java/Others)    Memory Limit: 131070/65535 K (Java/Others) Total Submission(s): 19592    Accepted Submission(s): 8592 Problem Description Ignatius最近遇到一个难题,老师交给他很多单词(只有小写字母组成,不会有重复的单词出现),现在老师要他统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀).   Input 输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是老师交给Ignatius统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串. 注意:本题只有一组测试数据,处理到文件结束.   Output 对于每个提问,给出以该字符串为前缀的单词的数量.   Sample Input banana band bee absolute acm ba b band abc   Sample Output 2 3 1 0

说明：用C++提交超时，用G++可以AC.

#include<stdio.h>
#include<string.h>
#include<malloc.h>
#define change(n) (n-('a'-0))
#define NUM 26
#define MAX 20
struct Trie 
{
	int num;//该节点是几个单词的公共部分
	struct Trie *next[NUM];
};
void Build(struct Trie *root,char *c)
{
	int i,j,n,m;
	struct Trie *p=root;
	n=strlen(c);
	for(i=0;i<n;i++)
	{
		m=change(c[i]);
		if(p->next[m])
		{
			p=p->next[m];
			p->num+=1;
			continue;
		}
		//-----创立新节点-----// 
		struct Trie *s;
		s=(struct Trie *)malloc(sizeof(struct Trie));
		s->num=0;
		for(j=0;j<NUM;j++)
		s->next[j]=NULL;
		//--------------------//
		p->next[m]=s;
		
		p=p->next[m];
		p->num+=1;
	}
}
int Find(struct Trie *root,char *s)
{
	int i,j,n,m;
	struct Trie *p=root;
	n=strlen(s);
	for(i=0,j=0;i<n;i++)
	{
		m=change(s[i]);
		if(p->next[m])
		{
		   p=p->next[m];
		   continue;
		}
		if(!p->next[m])
		{
			j=-1;
		    break;
	    }
	}
	if(j==-1) return 0;
	else return p->num;
}
void Delete(struct Trie *root)
{
	int i;
	for(i=0;i<NUM;i++)
	{
		if(root->next[i])
		Delete(root->next[i]);
	}
	free(root);
}
int main()
{
	char c[MAX],s[MAX];
	int n,i;
	
	//-----创建根节点-----//
	struct Trie *head;
	head=(struct Trie *)malloc(sizeof(struct Trie));
	head->num=0;
	for(i=0;i<NUM;i++)
	head->next[i]=NULL;
	//--------------------// 
	while(gets(c)&&c[0])//以换行结束要这样结束 
	{
		Build(head,c);
	}//建立Trie树成功 
	while(scanf("%s",s)!=EOF)
	{
		n=Find(head,s);
		printf("%d\n",n);
	}
	Delete(head); 
	
	return 0;
}