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

来源：互联网发布：wamp mysql 乱码编辑：程序博客网时间：2024/04/29 22:52

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

解题思路：很明显的一道字典树的入门题。一直都习惯用数组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;}

0 0