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;
}