hdu 1251 （Trie 树）

Ignatius最近遇到一个难题,老师交给他很多单词(只有小写字母组成,不会有重复的单词出现),现在老师要他统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀).
Input
输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是老师交给Ignatius统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串.

注意:本题只有一组测试数据,处理到文件结束.
Output
对于每个提问,给出以该字符串为前缀的单词的数量.
Sample Input
banana
band
bee
absolute
acm

ba
b
band
abc
Sample Output
2
3
1
0
用g++ 超内存， 用c++不会

#include <iostream>
#include <cstring>
#include <cstdio>
#include <string>
using namespace std;
const int sonnum=26,base='a';
char s1[12],ss[12];

struct Trie
{
    int num;
    bool flag;
    struct Trie *son[sonnum];
    Trie()
    {
        num=1,flag=false;
        memset(son,0,sizeof(son));
    }
};

Trie *NewTrie()
{
    Trie *temp=new Trie;
    return temp;
}

void Insert(Trie *root,char *s)
{
    Trie *temp=root;
    while(*s){
        if(temp->son[*s-base]==NULL)
        {
            temp->son[*s-base]=NewTrie();
        }
        else 
            temp->son[*s-base]->num++;
        temp=temp->son[*s-base];
        s++;
    }
    temp->flag=true;
}

int search(Trie *root,char *s)
{
    Trie *temp=root;
    while(*s)
    {
        if(temp->son[*s-base]==NULL) return 0;
        temp=temp->son[*s-base];
        s++;
    }
    return temp->num;
}

int main()
{
    Trie *root=NewTrie();
    root->num=0;
    while(gets(s1)&&strcmp(s1,"")!=0)
    {
        Insert(root,s1);
    }
    while(cin>>ss)
    {
        int ans=search(root,ss);
        cout<<ans<<endl;
    }
    return 0;
}