hdu1251

http://acm.hdu.edu.cn/showproblem.php?pid=1251

今天刚看了字典树的资料，想做题练练手，结果很悲剧，字典树调了4个小时才做出一道水题

此题题意就是求解单词前缀的数量，通过字典树减少内存，并且查找单词时间只与单词长度有关

对于此题主要有一点提示就是，题目明确表示输入不会有相同单词，就是说查找单个字母若已存在字典树中只需将它的前缀数+1即可，而不存在的话就是新开一个节点，前缀数为1

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
 struct node
{
    node*next[26];
    int num;
}*root;
int bulid(char str[],int k,node*tmproot)
{
    do
    {
        if(tmproot->next[str[k]-'a']==NULL)
        {
            node*p=(node*)malloc(sizeof(node));
            tmproot->next[str[k]-'a']=p;
            for(int i=0;i<26;i++)
            tmproot->next[str[k]-'a']->next[i]=NULL;
            tmproot->next[str[k]-'a']->num=1;
            tmproot=tmproot->next[str[k]-'a'];
        }
        else
        {
            tmproot->next[str[k]-'a']->num+=1;
            tmproot=tmproot->next[str[k]-'a'];
        }
    }while(++k<strlen(str));                                       //构建字典树，并存储前缀编码数
}
int visit(char str_visit[],int k,node*tmproot)
{
    while(k<strlen(str_visit))
     if(tmproot->next[str_visit[k]-'a']!=NULL)
      tmproot=tmproot->next[str_visit[k++]-'a'];
     else return 0;
    return tmproot->num;                                           //在遍历字符过程中如果字符不存在则返回0，存在返回最后一个字符的前缀编码值
}
int main()
{
    char str[20];
    root=(node*)malloc(sizeof(node));
    for(int i=0;i<26;i++) root->next[i]=NULL;
    while(gets(str)&&str[0]!='\0')
    {
        bulid(str,0,root);
        memset(str,'\0',sizeof(str));
    }
    while(scanf("%s",str)!=EOF)
     printf("%d\n",visit(str,0,root));
    return 0;
}