HDU 1251 统计难题（字典树模板题）

http://acm.hdu.edu.cn/showproblem.php?pid=1251

题意：
给出一些单词，然后有多次询问，每次输出以该单词为前缀的单词的数量。

思路：

字典树入门题。

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 using namespace std;
 const int maxn = ;

 int num = ;

 struct Trie
 {
     int son[];
     int cnt;  //前缀数量
     int ends; //单词数量，在本题中其实并没有用到
 }t[maxn];

 void init(int x)
 {
     t[x].ends = ;
     t[x].cnt = ;
     memset(t[x].son,,sizeof(t[x].son));
 }

 void insert(char* s)
 {
     int u = , n = strlen(s);
     for(int i=;i<n;i++)
     {
         int c = s[i]-'a';
         if(!t[u].son[c])
         {
             num++;
             init(num);
             t[u].son[c] = num;
         }
         u = t[u].son[c];
         t[u].cnt++;
     }
     t[u].ends++;
 }

 int query(char* s)
 {
     int u = , n = strlen(s);
     for(int i=;i<n;i++)
     {
         int c = s[i]-'a';
         if(t[u].son[c] == )  return ;
         u = t[u].son[c];
     }
     return t[u].cnt;
 }

 char s[];

 int main()
 {
     while(gets(s) && strcmp(s,"")!=)  insert(s);
     while(scanf("%s",s)!=EOF)  printf("%d\n",query(s));
     return ;
 }