字典树：求以某字符串开始的单词个数

http://hihocoder.com/contest/hiho2/problem/1

与之前做的有点像：http://blog.csdn.net/u011644423/article/details/37833905

 

稍微改下，注意每次在建立Trie树时同时进行统计

#include <stdio.h>  
#include<stdlib.h>  
#define MAX 26  
//using namespace std;   
typedef struct TrieNode                     //Trie结点声明   
{  
    //bool isStr;      
    int count;  
    //标记该结点处是否构成单词 可能大于一 
    struct TrieNode *next[MAX];            //儿子分支   
}Trie;  
   
void insert(Trie *root,const char *s)     //将单词s插入到字典树中   
{  
    if(root==NULL||*s=='\0')  
        return;  
    int i;  
    Trie *p=root;  
    while(*s!='\0')  
    {  
        if(p->next[*s-'a']==NULL)        //如果不存在，则建立结点   
        {  
            Trie *temp=(Trie *)malloc(sizeof(Trie));  
            for(i=0;i<MAX;i++)  
            {  
                temp->next[i]=NULL;
                  
            }  
           // temp->isStr=false;  
            p->next[*s-'a']=temp;  
            p=p->next[*s-'a'];    
            p->count=1;
        }     
        else  
        {  
            p=p->next[*s-'a'];  
            (p->count)++;  
        }  
        s++;  
    }  
   // p->isStr=true;                       //单词结束的地方标记此处可以构成一个单词   
}  

int findmin(Trie *root,char *str){  
Trie *p=root;  
int num=0;
while(*str!='\0'){  
    if(p->next[*str-'a']==NULL){  
       num=0; break;  
    }  

    num=(p->next[*str-'a'])->count;
//    printf("%c",*str);  
    p=p->next[*str-'a'];  
    str++;  
  
}  
  return num;
}  
void del(Trie *root)                      //释放整个字典树占的堆区空间   
{                                           //递归  
    int i;  
    for(i=0;i<MAX;i++)  
    {  
        if(root->next[i]!=NULL)  
        {  
            del(root->next[i]);  
        }  
    }  
    free(root);  
}  
   
int main(int argc, char *argv[])  
{  
    int i,j;  
  //  int n,m;                              //n为建立Trie树输入的单词数，m为要查找的单词数   
     
    Trie *root= (Trie *)malloc(sizeof(Trie));  
    for(i=0;i<MAX;i++)  
    {  
        root->next[i]=NULL;  
    }  
    //root->isStr=false;  
    //scanf("%d",&n);  
    //getchar();  
   
    int n;
    scanf("%d",&n);
    for(j=1;j<=n;j++){
        getchar();  
        char s[12];
        scanf("%s",s);
        insert(root,s);     
    }
//    while(scanf("%s",s[gsum])!=EOF)  
//        //scanf("%s",s);  
//    {  
//          
//        insert(root,s[gsum]);  
//        getchar();  
//        gsum++;  
//    }  
     int m;
    scanf("%d",&m);
    
    for(j=1;j<=m;j++){
        getchar();  
        char s2[12];
        scanf("%s",s2);
        printf("%d\n",findmin(root,s2));   
    }
     
        
      
   // del(root);                         //释放空间很重要   
    return 0;  
}  

 

随便说下。 用博客园写博客比csdn爽太多