统计难题——Trie树

题目描述：
题目链接： HDU 1251 http://acm.hdu.edu.cn/showproblem.php?pid=1251
Ignatius 最近遇到一个难题，老师交给他很多单词(只有小写字母组成，不会有重复的单词出现)，现在老师要他统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀)。
输入格式：
输入数据的第一部分是一张单词表（不超过 $10^4$ 个单词），每行一个单词，单词的长度不超过 10，它们代表的是老师交给 Ignatius 统计的单词，一个空行代表单词表的结束。第二部分是一连串的提问，每行一个提问，每个提问都是一个字符串（不超过 $10^4$ 个提问串）。
注意:本题只有一组测试数据，处理到文件结束。
输出格式：
对于每个提问，给出以该字符串为前缀的单词的数量。
样例输入
banana
band
bee
absolute
acm

ba
b
band
abc
样例输出：
2
3
1
0
题目分析：
trie树模板。只需建树时记一下个数。
附代码：

#include<iostream>
#include<cstring>
#include<string>
#include<cstdio>
#include<cstdlib>
#include<ctime>
#include<queue>
#include<iomanip>
#include<cmath>
#include<set>
#include<cctype>
#include<map>
#include<algorithm>
using namespace std;

struct node{
    int son[26];
    int num;
}trie[100005];
char s[15];
int ans,tot;

void create()
{
    int len,u=0;
    len=strlen(s);
    for(int i=0;i<len;i++)
    {
        if(trie[u].son[s[i]-'a']==0)
            trie[u].son[s[i]-'a']=++tot;
        u=trie[u].son[s[i]-'a'];
        trie[u].num++;
    }
}

int getans()
{
    int len,u=0;
    len=strlen(s);
    for(int i=0;i<len;i++)
    {
        if(trie[u].son[s[i]-'a']==0)
            return 0;
        u=trie[u].son[s[i]-'a'];
    }
    return trie[u].num;
}

int main()
{
    //freopen("lx.in","r",stdin);

    while(gets(s)&&s[0]!='\0')
        create();
    while(scanf("%s",s)!=EOF)
    {
        ans=getans();
        printf("%d\n",ans);
    }

    return 0;
}