[hdu1251]统计难题(trie模板题)

题意：返回字典中所有以测试串为前缀的字符串总数。

解题关键：trie模板题，由AC自动机的板子稍加改造而来。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<iostream>
#include<cmath>
#include<queue>
using namespace std;
typedef long long ll;
const int N=;
const int MAXN=;
struct Trie{//数组形式
    int Next[MAXN][N],Fail[MAXN],End[MAXN],root,tot;//大小为所以匹配字符串的总和
    int newnode(){//结构体内部用
        for(int i=;i<N;i++) Next[tot][i]=-;
        End[tot++]=;
        return tot-;
    }
    void init(){
        tot=;
        root=newnode();
    }
    void insert(char buf[]){
        int len=strlen(buf);
        int now=root;//now是temp指针
        for(int i=;i<len;i++){
            int k=buf[i]-'a';
            if(Next[now][k]==-)  Next[now][k]=newnode();//next数组代表的是下一个字符索引
            now=Next[now][k];
            End[now]++;
        }
    }
    int fnd(char buf[]){
        int len=strlen(buf);
        int now=root;//now是temp指针
        for(int i=;i<len;i++){
            int k=buf[i]-'a';
            if(Next[now][k]==-)  return ;//next数组代表的是下一个字符索引
            now=Next[now][k];
        }
        return End[now];
    }
};
Trie ac;
char buf[];
int main(){
    ac.init();
    while(gets(buf)){
        if(buf[]==NULL)break;//gets读入的回车会自动转化为NULL
        ac.insert(buf);
    }
    while(gets(buf)){
        printf("%d\n",ac.fnd(buf));
    }
    return ;
}