HDU 1251 统计难题 （裸的字典树）

题目链接

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

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

注意:本题只有一组测试数据,处理到文件结束.

Output
对于每个提问,给出以该字符串为前缀的单词的数量.

Sample Input
banana
band
bee
absolute
acm

ba
b
band
abc

Sample Output
2
3
1
0

分析：

题意：首先会给定一系列的单词，当输入的单词时空格的时候结束，然后给定前缀，让输出前面给定的单词中以这个前缀为前缀的单词由多少个（包括单词本身是前缀）。

这道题就是一道简单的字典树的入门题，一个纯粹的裸字典树，用数组来模拟字典树，在建树的同时保存单词的前缀的个数。

代码：

#include<stdio.h>
#include<iostream>
#include<string.h>
using namespace std;
int tree[400000][27];//数组模拟的字典树
int val[400000];//查询每次到这个地方的单词前缀由多少个
int falg;//作为一个标记来唯一的表示某个位置上的一个数字
void init()
{
    memset(tree,0,sizeof(tree));
    memset(val,0,sizeof(val));
    falg=1;
}

void build(char s[])//构建字典树
{
    int len=strlen(s);
    int u=0;
    for(int i=0;i<len;i++)
    {
        int j=s[i]-'a';//将字母转换成数字
        if(tree[u][j]==0)//如果当前的这个单词没有出现过
        {
            tree[u][j]=falg++;//就给这个位置标记已经出现过了单词
        }
        u=tree[u][j];//跳转到这个位置接着往下
        val[u]++;//标记到目前为止的这个前缀出现的次数
    }
}
int query(char s[])
{
    int len=strlen(s);
    int u=0;
    for(int i=0;i<len;i++)
    {
        int j=s[i]-'a';
        if(tree[u][j]==0)//相当于这个前缀没有出现过
            return 0;
        else
            u=tree[u][j];//接着往下找
    }
    return val[u];
}
int main()
{
    init();
    char s[15];
    while(gets(s))
    {
        if(strlen(s)==0) break;
        build(s);
    }
    while(gets(s))
    {
        if(strlen(s)==0) break;
        printf("%d\n",query(s));
    }
    return 0;
}