统计难题

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

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

题解：Trie模板题。在插入每个单词的时候统计前缀的个数即可。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#define maxn 1100000
using namespace std;
struct Nod {        //0为无效值
    int lnk[26], val;
    void init() {
        memset(lnk, 0, sizeof(lnk));
        val = 0;
    }
};
const char BASE = 'a';
struct Trie {
    Nod buf[maxn];
    int len;
    void init() {
        buf[len=0].init();
    }
    void insert(char * str) {
        int now = 0;
        for(int i = 0; str[i]; i ++) {
            int & nxt = buf[now].lnk[str[i]-BASE];
            if(!nxt)buf[nxt=++len].init();
             now = nxt;
             buf[now].val++;
        }
    }
    int search(char * str) {
        int now = 0;
        for(int i = 0; str[i]; i ++) {
            int & nxt = buf[now].lnk[str[i]-BASE];
            if(!nxt)    return 0;
            now = nxt;
        }
        return buf[now].val;
    }
} trie;
char str[100];
 int main(){
     int n,m;
   // scanf("%d",&n);
    trie.init();

    while(gets(str),strcmp(str,""))
    {
        trie.insert(str);     //先建立字典树
    }
    while(gets(str))
    {
        printf("%d\n", trie.search(str));
    }

 }