HDU - 1251 统计难题（字典树水题）

统计难题

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

Input

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

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

Output

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

Sample Input

banana
band
bee
absolute
acm

ba
b
band
abc

Sample Output

2
3
1
0

字典树入门。 对节点添加一个 Sz 数组，到达这个节点的次数。询问的时候返回Sz值即可。

AC代码：

#include<bits/stdc++.h>
#define debug(x) cout << "[" << #x <<": " << (x) <<"]"<< endl
#define pii pair<int,int>
#define clr(a,b) memset((a),b,sizeof(a))
#define rep(i,a,b) for(int i = a;i < b;i ++)
#define pb push_back
#define MP make_pair
#define LL long long
#define ull unsigned LL
#define ls i << 1
#define rs (i << 1) + 1
#define INT(t) int t; scanf("%d",&t)

using namespace std;

const int maxn = 1e6 + 10;
int trie[maxn][26];
int Sz[maxn];
int k = 1;      ///根节点为 0

void add(char *s){
    int len = strlen(s);
    int p = 0;
    for(int i = 0;i < len;++ i){
        int c = s[i] - 'a';
        if(!trie[p][c]){
            trie[p][c] = k ++;
        }
        p = trie[p][c];
        ++ Sz[p];
//    debug(Sz[p]);
//    debug(p);
    }
//    ++ Sz[p];
}

int query(char *s){
    int p = 0,len = strlen(s);
    for(int i = 0;i < len;++ i){
        int c = s[i] - 'a';
        if(!trie[p][c]) return 0;
        p = trie[p][c];
    }
    return Sz[p];
}

int main() {
    char s[15];
    int flag = 0; ///正在读入单词表
    while(gets(s)){
//        debug(s);
//        debug(strlen(s) == 0);
        if(strlen(s) == 0){
            flag = 1;
            continue;
        }
        if(!flag){
            add(s);
        }
        else {
            printf("%d\n",query(s));
        }
    }
    return 0;
}