數據結構之字典樹 HDU 1251

-----原來好久之前就做了，結果後來忘了，唉，真是要多練啊

統計難題

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

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 dictree   // 結構體定義樹 
{   
    struct dictree *child[26];  //結構體指針數組，一棵樹有26個孩子 
    int n;   
};   
struct dictree *root;   

void insert (char *source)   
{   
    int len,i,j;   
    struct dictree *current,*newnode;   
    len=strlen(source);      
    current=root;   
    for(i=0;i<len;i++)   
    {   
        if(current->child[source[i]-'a']!=NULL)   
        {
            current=current->child[source[i]-'a'];   //將current 的指針移動到其子孫處 
            current->n=current->n+1; 
        }
else   
        {   
            newnode=(struct dictree *)malloc(sizeof(struct dictree));   
            for(j=0;j<26;j++)   
                newnode->child[j]=0; 
            current->child[source[i]-'a']=newnode;   
            current=newnode;
            current->n=1;    //
        }   
    }   
}   
int find(char *source)   
{   
    int i,len;   
    struct dictree *current;  
    len=strlen(source);
    if(len==0) return 0;   
    current=root;   
    for(i=0;i<len;i++)   
    {   
      if(current->child[source[i]-'a']!=0)   
      current=current->child[source[i]-'a'];   
      else return 0;   
    }   
    return current->n;   
}   
int main()
{
    char temp[11];
    int i,j;
    root=(struct dictree *)malloc(sizeof(struct dictree));   
    for(i=0;i<26;i++)   
        root->child[i]=0;   
    root->n=2; 
    while(gets(temp),strcmp(temp,"")!=0)
        insert(temp);
    while(scanf("%s",temp)!=EOF)
    {
        i=find(temp);
        printf("%d\n",i);
    }
}