HDU 1251_统计难题

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<stdio.h>  
    #include<string.h>  
    #include<stdlib.h>  
    struct node
	{
		struct node *next[26];
	    int sum;
	};
	 
	struct node *root = NULL;
	 
	void find(char temp[])
	{
	    int i,len;
	    struct node *pre=NULL,*now=NULL;
	 
	    now = pre = root;
	    len = strlen(temp);
	 
	    for(i=0;i<len;i++)
	    {
	        if(now->next[temp[i]-'a']!=NULL)
	         now = now->next[temp[i]-'a'];
	        else
	         break;
	    }
	 
	    if(i>=len)
	      printf("%d\n",now->sum);
	    else
	      printf("0\n");
	}
	 
	void insert(char * temp)
	{
	    int i,j,len;
	 
	    struct node *newnode=NULL,*now=NULL;
	 
	    now=root;
	    len = strlen(temp);
	 
	    for(i=0;i<len;i++)
	    {
	        if(now->next[temp[i]-'a']!=NULL)
	        {
            now=now->next[temp[i]-'a'];
	            now->sum++;
	        }
        else
	        {
	            newnode=(struct node*)malloc(sizeof(struct node));
	 
	            for(j=0;j<26;j++)
            {
                newnode->next[j]=NULL;
	            }
	 
	            newnode->sum=0;
	 
           now->next[temp[i]-'a'] = newnode;
	           newnode->sum++;
           now = newnode;
        }
	 
	    }
	}
	 
	int main()
	{
	    int i;
	    char str[15];
	 
	    root=(struct node *)malloc(sizeof(struct node));
	 
	    for(i=0;i<26;i++)
    {
	        root->next[i]=NULL;
	    }
	 
	    root->sum=0;
	 
	   
	 
	    while(gets(str)&&str[0]!='\0')
	    {
	        insert(str);
	    }
	    while(scanf("%s",str)==1)
	    {
	        find(str);
	    }
 
	    return 0;
	}

解题思路：利用字典树建立数据结构，找相同的前缀。。。
字典树的优点是：利用字符串的公共前缀来减少查询时间，最大限度地减少无谓的字符串比较，查询效率比哈希表高。