字符串_字典树(模板 hdu 1251)

例题链接：http://acm.hdu.edu.cn/showproblem.php?pid=1251
代码：

#include "stdio.h" // 字典树模板题  hdu 1251
#include "string.h"
#include "stdlib.h"

struct node{
    int num;
    struct node *next[26];
};

struct node *root;

void Insert(char *k)
{
    int i;
    struct node *a = root;
    while(k[0]!='')
    {
        if(a->next[k[0]-'a'] == NULL)
        {
            node *tt;
            tt = (node *)malloc(sizeof(node));
            tt->num = 1;
            for(i=0;i<26;i++) tt->next[i] = NULL;
            a->next[k[0]-'a'] = tt;
            a = tt;  //移向下一级
        }
        else
        {
            a = a->next[k[0]-'a'];  //移向下一级
            a->num = a->num + 1;
        }
        k++;
    }
}

int Find(char *k)
{
    int ans;
    struct node *a = root;
    while(k[0]!='')
    {
        if(a->next[k[0]-'a']==NULL)
            return 0;
        else
        {
            ans = a->next[k[0]-'a']->num;
            a = a->next[k[0]-'a'];
        }
        k++;
    }
    return ans;
}

void BFS(node *k);

int main()
{
    int i;
    char str[15];
    root = (node *)malloc(sizeof(node));  //给root指针开辟空间
    root->num = 0;
    for(i=0;i<26;i++)  //root下的指针初始化
        root->next[i] = NULL;
    while(gets(str) && strcmp(str,"")!=0)
        Insert(str);
    while(scanf("%s",str)!=-1)
        printf("%d
",Find(str));
    BFS(root);
    return 0;
}

void BFS(node *k)  //深搜去释放内存！
{
    int i;
    if(k==NULL)    return ;
    for(i=0;i<26;i++)
    {
        if(k->next[i]!=NULL)
            BFS(k->next[i]);
    }
    free(k);
}