（trie）HDU1251 统计难题

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

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

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

banana
band
bee
absolute
acm

ba
b
band
abc

Sample Output

2
3
1
0

完完全全的trie裸题，不过要注意数组（或map）的大小，防止MLE。

#include <iostream>
#include <string>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <queue>
#include <set>
#include <map>
#include <list>
#include <vector>
#include <stack>
#define mp make_pair
//#define P make_pair
#define MIN(a,b) (a>b?b:a)
//#define MAX(a,b) (a>b?a:b)
typedef long long ll;
typedef unsigned long long ull;
const int MAX=1e6+5;
const int MAX_V=1e3+5;
const ll INF=4e18+5;
const double M=4e18;
using namespace std;
const int MOD=1e9+7;
typedef pair<ll,int> pii;
const double eps=0.000000001;
#define rank rankk
map<int,int>ch[MAX];
int val[MAX];
struct Trie
{
    int num;
    Trie(){num=1;}
    int idx(char x)
    {
        return x-'a';
    }
    void clear()
    {
        num=1;
    }
    void insert(char *s)
    {
        int u=0,len=strlen(s);
        for(int i=0;i<len;i++)
        {
            int c=idx(s[i]);
            if(!ch[u][c])
            {
                val[num]=0;
                ch[u][c]=num++;
            }
            u=ch[u][c];
            ++val[u];
        }
    }
    int check(char *s)
    {
        int u=0,len=strlen(s);
        for(int i=0;i<len;i++)
        {
            int c=idx(s[i]);
            if(!ch[u][c])
                return 0;
            u=ch[u][c];
        }
        return val[u];
    }
};
char a[300];
int main()
{
    Trie dic;
    while(1)
    {
        gets(a);
        if(strlen(a)==0)
            break;
        dic.insert(a);
    }
    while(scanf("%s",a)!=EOF)
    {
        printf("%d
",dic.check(a));
    }
}