HDU1251 统计难题 tire树

题目大意：给你一系列字符串作为母串，对于若干次询问，每次询问包含一个字符串，输出该以字符串为前缀的母串的数目。

分析：典型的tire树，在每次插入数据的时候纪录前缀数。

实现代码如下：

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
using namespace std;
#define son_num 30 //字符串中包含的字符个数
#define maxn 11 //单词的最大长度
struct tire
{
    int num;//纪录到达该节点的字符串有多少个
    bool terminal; //如果terminal=true,表示当前点为该字符串最后一个字符
    struct tire *next[son_num]; //纪录下一个节点
};
void insert(tire *root,char str[]) //插入操作
{
    int i=0;
    tire *p=root;
    while(str[i]!='\0')
    {
        if(!p->next[ str[i]-'a' ]) //如果不存在，建立新节点
        {
            tire *q=(tire *)malloc(sizeof(tire));
            for(int i=0;i<son_num;i++)
              q->next[i]=NULL;
            q->terminal=false;
            q->num=0;

            p->next[ str[i]-'a' ]=q;
        }
        p=p->next[ str[i]-'a' ];
        p->num++;
        i++;
    }
    p->terminal=true;
}
int find(tire *p,char str[]) //查找操作
{
    int i=0;
    while(str[i]!='\0'&&p->next[ str[i]-'a' ])
    {
        p=p->next[ str[i]-'a' ];
        i++;
    }
    if(str[i]=='\0') return p->num;
    return 0;
}
void del(tire *root) //清空操作
{
    for(int i=0;i<son_num;i++)
      if(root->next[i]!=NULL)
        del(root->next[i]);
    free(root);
}
int main()
{
    tire *root=(tire *)malloc(sizeof(tire)); //定义根结点
    for(int i=0;i<son_num;i++)
      root->next[i]=NULL;
    root->terminal=false;
    root->num=0;
    
    char str[maxn];
    while(gets(str)) //建立字典树
    {
        if(strcmp(str,"")==0) break;
        insert(root,str);
    }
    while(gets(str)) //查找操作
    {
        if(strcmp(str,"")==0) break;
        printf("%d\n",find(root,str));
    }
    del(root); //释放字典树占用的空间
    return 0;
}