hdu 3065 AC自动机

//    hdu 3065 AC自动机
//
//    题目大意：
//
//        给你n个短串，然后给你一个长串，问：各个短串在长串中，出现了多少次
//
//    解题思路:
//
//        AC自动机，插入，构建，查询就OK啦
//
//    感悟：
//
//        这道题真的是1A的哟~~~很开心~~~尽管是个裸地，继续加油哟~~~FIGHTING

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <queue>
using namespace std;

const int MAX_NODE = ;
const int SIGMA = ;

struct Aho_Corasick{

    int ch[MAX_NODE][SIGMA];
    int f[MAX_NODE];
    int val[MAX_NODE];
    int last[MAX_NODE];
    int cnt[];
    int sz;

    void init(){
        sz = ;
        memset(ch[],,sizeof(ch[]));
        memset(cnt,,sizeof(cnt));
        val[] = ;
    }

    void insert(char *s, int v){
        int u = ;
        int n = strlen(s);
        for (int i=;i<n;i++){
            int c = s[i];
            if (!ch[u][c]){
                memset(ch[sz],,sizeof(ch[sz]));
                val[sz] = ;
                ch[u][c] = sz++;
            }
            u = ch[u][c];
        }
        val[u] = v;
    }

    void getfail(){
        queue<int> que;
        for (int c = ;c < SIGMA; c++){
            int u = ch[][c];
            if(u){
                que.push(u);
                f[u] = ;
                last[u] = ;
            }
        }

        while(!que.empty()){
            int r = que.front();
            que.pop();

            for (int c =  ; c < SIGMA;c++){
                int u = ch[r][c];

                if (!u)
                    continue;

                que.push(u);

                int v = f[r];

                while( v && !ch[v][c])
                    v = f[v];

                f[u] = ch[v][c];

                last[u] = val[f[u]] ? f[u] : last[f[u]];
            }
        }
    }

    void get_cnt(int u){
        if (u){
            cnt[val[u]]++;
            get_cnt(last[u]);
        }
    }

    void query(char *s){
        int u = ;

        int n = strlen(s);

        for(int i=;i<n;i++){
            int c = s[i];

            while(u && !ch[u][c])
                u = f[u];

            u = ch[u][c];

            if (val[u]){
                get_cnt(u);
            }else if (last[u]){
                get_cnt(last[u]);
            }
        }
    }

}ac;

int n;

char p[][];

char str[];

void input(){
    ac.init();
    char s[];
    for (int i=;i<=n;i++){
        scanf("%s",s);
        memcpy(p[i],s,sizeof(s));
        ac.insert(s,i);
    }
    ac.getfail();

    scanf("%s",str);

    ac.query(str);
    for (int i = ;i<=n;i++){
        if (ac.cnt[i]){
            printf("%s: %d\n",p[i],ac.cnt[i]);
        }
    }
}

int main(){
    //freopen("1.txt","r",stdin);
    while(scanf("%d",&n)!=EOF){
        input();
    }
    return ;
}