hdu-1251 统计难题---字典树

题目链接：

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

题目大意：

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

 解题思路：

将每个单词存入字典树中，直接查询前缀出现次数即可，在加入字典树的时候，在每个字母表示的边上后面的节点数加1，记录前缀出现次数。

传送门：字典树模板

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<map>
#include<set>
#include<cmath>
#include<algorithm>
#include<vector>
#include<sstream>
#define lowbot(i) (i&(-i))
using namespace std;

const int maxn = 1e6 + 10;
int tree[maxn][26];
//字典树tree[u][v]表示编号为u的节点的下一个字母为v连接的节点的编号
int idx(char c){ return c - 'a'; }//可以写成宏定义
int tot = 1;//根节点编号为1
int sum[maxn];//标记以该节点结束的前缀出现次数
void Insert(char s[], int u)//u表示根节点
//插入字符串s
{
    for(int i = 0; s[i]; i++)
    {
        int c = idx(s[i]);
        if(!tree[u][c])
            tree[u][c] = ++tot;
        sum[tree[u][c]]++; //前缀后面的那个节点数目加一
        u = tree[u][c];
    }
}

int Find_sum(char s[], int u)
{
    for(int i = 0; s[i]; i++)
    {
        int c = idx(s[i]);
        if(!tree[u][c])return 0;
        u = tree[u][c];
    }
    return sum[u];//返回最后一个字母表示的边连接的后面那个节点，所记录的sum值
}
int main()
{
    char s[15];
    while(gets(s) && strlen(s))Insert(s, 1);
    while(scanf("%s", s) != EOF)
    {
        printf("%d\n", Find_sum(s, 1));
    }
    return 0;
}