python/TangDou/LuoGuBook/P3808.cpp

#include <bits/stdc++.h>

using namespace std;

const int N = 10010 * 55;    //模式串最长长度,短串
const int M = 1e6 + 10; //长度为m的文章，长串

int n;
//tr:trie树，每个结点最多26个儿子
//cnt:trie的每个结点存在的以此结点结尾的字符串个数
//idx:游标变量，结点号
int tr[N][26], cnt[N], idx;
int st[N];      //是不是统计过了，统计过的不用重复统计
string str;     //字符串变量，前面用做输入模式串，最后一个是文章串
//AC自动机自己的数据结构 q:bfs用的队列 fail:失配指针
int q[N], fail[N];

//构建Trie树
void insert() {
    int p = 0;                              //游标变量
    for (int i = 0; i < str.size(); i++) {  //遍历模式字符串
        int t = str[i] - 'a';               //计算对应的索引号
        if (!tr[p][t]) tr[p][t] = ++idx;    //如果不存在这个结点，则创建之
        p = tr[p][t];                       //走进去
    }
    cnt[p]++;                               //打上字符串完结标识
}

//构建AC自动机(填充失配指针)
void build() {
    int hh = 0, tt = -1;//清空队列，所以前面不需要每次都memset清空q
    for (int i = 0; i < 26; i++)    //查找root的所有26个可能存在的儿子
        if (tr[0][i])             //如果该儿子存在
            q[++tt] = tr[0][i];   //把这个儿子放入队列中

    while (hh <= tt) {          //bfs框架
        int p = q[hh++];        //取出队列头,父结点
        for (int i = 0; i < 26; i++) {  //尝试找出该结点的所有儿子
            int c = tr[p][i];   //子结点
            //如果不存在,这个的失配指针指向,父节点的失配指针对应的相同字节点下
            if (!c) tr[p][i] = tr[fail[p]][i];
            else {
                //存在依旧指向父节点的失配指针下的同子节点
                fail[c] = tr[fail[p]][i];
                q[++tt] = c;//放入队列
            }
        }
    }
}

int main() {
    cin >> n;
    for (int i = 1; i <= n; i++) {
        cin >> str;
        //构建Trie树
        insert();
    }
    //构建AC自动机
    build();
    //读入文章长串
    cin >> str;
    int res = 0;
    //j记录当前树节点的指针，初始是根节点
    for (int i = 0, j = 0; i < str.size(); i++) {  //枚举总串str的每一个字母
        int u = str[i] - 'a';
        j = tr[j][u];       //跳到下一个树节点
        int p = j;          //每次从当前树节点开始

        while (p && !st[p]) {
            res += cnt[p];  //累加命中的模式串个数
            cnt[p] = 0;     //去除标记
            p = fail[p];    //继续查询
            st[p] = 1;      //标识已统计过
        }
    }
    printf("%d\n", res);
    return 0;
}