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

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

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

思路：

字典树入门题。

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

int num = 0;

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

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

void insert(char* s)
{
    int u = 0, n = strlen(s);
    for(int i=0;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 = 0, n = strlen(s);
    for(int i=0;i<n;i++)
    {
        int c = s[i]-'a';
        if(t[u].son[c] == 0)  return 0;
        u = t[u].son[c];
    }
    return t[u].cnt;
}

char s[15];

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