HDU 1251 (初级字典树)

Problem Description
Ignatius最近遇到一个难题,老师交给他很多单词(只有小写字母组成,不会有重复的单词出现),现在老师要他统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀).

Input
输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是老师交给Ignatius统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串.

注意:本题只有一组测试数据,处理到文件结束.

思路：

就是构建+查找
构建字典树的时候，注意num 的用处和flag的用处，分别代表着单词的数目和结尾
标志。

#include <iostream>#include <cstdio>#include <cstring>using namespace std;const int base = 'a',maxn = 26;char s1[12],s2[12];struct Tire{    int num;    int flag;    Tire *son[maxn];    Tire() {        //注意初始化        num = 1;        flag = false;        memset(son,NULL,sizeof(son));    }};/*-------------构造字典树------------*/Tire* NewTire(){    Tire *temp = new Tire;    return temp;}void Insert(Tire *root,char *s){    Tire *temp = root;    while(*s) {        if(temp->son[*s-base] == NULL) {            temp->son[*s-base] = NewTire();        }        else            temp->son[*s-base]->num++;        temp = temp->son[*s-base];        s++;    }    temp->flag = true;}int Serach(Tire *root,char *s){    Tire *temp = root;    while(*s) {        if(temp->son[*s-base] == NULL) return 0;        temp = temp->son[*s-base];        s++;    }    return temp->num;}int main(){    //freopen("in.txt","r",stdin);    Tire *root = NewTire();    root->num = 0;    while(gets(s1) && strcmp(s1,"") != 0) {        Insert(root,s1);    }    while(cin>>s2) {        int ans = Serach(root,s2);        cout<<ans<<endl;    }    return 0;}