HDU-#1251 统计难题（字典树）

      题目大意：给你很多单词，要求查找以某些单词为前缀的单词数量有多少？

     解题思路：很明显的一道字典树的入门题。一直都习惯用数组code的，突然看资料都是指针写的，很不习惯呀！不过重新看了下基础，回过头来写，发现还不是那么难，很好写的。言归正传，字典树主要就是那几步，一是建立树，查找和释放。资料在网上有很多的，就不赘述了。code给了很详细的解释了，详见code。

     题目来源：http://acm.hdu.edu.cn/showproblem.php?pid=1251

    code：

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

const int MAXN = 26;
typedef struct node{
    int cnt; //记录结点出现该词的次数
    node* next[26]; //存储结点的下一个字符
}Trie;

Trie root; //字典树的根节点

void createTrie(char* str){
    int len=strlen(str); //计算词的长度
    Trie *p=&root,*q; //初始化当前结点为根节点，以及中间变量的指针
    for(int i=0;i<len;i++){ //将每个字符依次插入字典树中
        int id=str[i]-'a'; //计算下标
        if(p->next[id]==NULL){ //如果没有出现过，则增加结点
            q=new Trie; //初始化一个新节点，开辟一个空间
            q->cnt=1; //将其出现次数初始化为1
            for(int j=0;j<MAXN;j++) //将其子结点初始化为NULL
                q->next[j]=NULL;
            p->next[id]=q; //将其付给p的下一个结点
            p=p->next[id]; //指针下移
        }
        else{ //如果出现过
            p->next[id]->cnt++; //将对应的子结点的计数加一
            p=p->next[id]; //指针下移
        }
    }
}

int findTrie(char* str){
    int len=strlen(str); //计算关键词的长度
    Trie *p=&root;  //初始化当前结点为根节点
    for(int i=0;i<len;i++){ //依次查找
        int id=str[i]-'a'; //计算下标
        p=p->next[id]; //下移指针查找
        if(p==NULL) return 0; //如果没有比较完成就不匹配，则说明该字典树中没有该词
    }
    return p->cnt; //如果遍历完成，存在则返回叶子结点的计数值
}

int main(){
    char str[11];
    for(int i=0;i<MAXN;i++)//初始化根节点所有子结点为空
        root.next[i]=NULL;
    while(gets(str) && str[0]!='\0') //接收词库
        createTrie(str);  //创建字典树
    memset(str,0,sizeof(str)); //将数组置0，接收查询键值
    while(scanf("%s",&str)!=EOF){ //接收查询关键字
        int ans=findTrie(str);
        printf("%d\n",ans);
    }
    return 0;
}