P3808 【模板】AC自动机（简单版）

题目描述

给定n个模式串和1个文本串，求有多少个模式串在文本串里出现过。

输入输出格式

输入格式：

第一行一个n，表示模式串个数；

下面n行每行一个模式串；

下面一行一个文本串。

输出格式：

一个数表示答案

输入输出样例

输入样例#1：

2
a
aa
aa

输出样例#1：

2

说明

subtask1[50pts]:$∑$length(模式串)$\le 10^6$,length(文本串)$\le10^6$,n=1；

subtask2[50pts]:$∑$length(模式串)$\le 10^6$,length(文本串)$\le 10^6$；

题解

存下板子。

yyb的板子跑得挺快的。

orz yyb

// luogu-judger-enable-o2
#include<bits/stdc++.h>
#define ll long long
using namespace std;
int qmax(int &x,int y) {if (x<y) x=y;}
int qmin(int &x,int y) {if (x>y) x=y;}
int read()
{
    char s;
    int k=0,base=1;
    while((s=getchar())!='-'&&s!=EOF&&!(isdigit(s)));
    if(s==EOF)exit(0);
    if(s=='-')base=-1,s=getchar();
    while(isdigit(s))
    {
        k=k*10+(s^'0');
        s=getchar();
    }
    return k*base;
}
void write(int x)
{
    if(x<0)
    {
        putchar('-');
        write(-x);
    }
    else
    {
        if(x/10)write(x/10);
        putchar(x%10+'0');
    }
}
const int maxn=1e6+100;
struct node
{
    int ch[26],end,fail;
} a[maxn];
int n,id;
queue<int> q;
char CH[maxn];
inline void build()
{
    int len=strlen(CH+1);
    int now=0;
    for (int i=1;i<=len;i++)
    {
        if (a[now].ch[CH[i]-'a']==0) a[now].ch[CH[i]-'a']=++id;
        now=a[now].ch[CH[i]-'a'];
    }
    a[now].end++;
}
void get_fail()
{
    for (int i=0;i<26;i++)
    {
        if (a[0].ch[i]!=0)
        {
            a[a[0].ch[i]].fail=0;
            q.push(a[0].ch[i]);
        }
    }
    while (!q.empty())
    {
        int u=q.front(),v;q.pop();
        for (int i=0;i<26;i++)
        {
            v=a[u].ch[i];
            if (v!=0)
            {
                a[v].fail=a[a[u].fail].ch[i];
                q.push(v);
            } else a[u].ch[i]=a[a[u].fail].ch[i];
        }
    }
}
int ac_query()
{
    int len=strlen(CH+1);
    int now=0,sum=0;
    for (int i=1;i<=len;i++)
    {
        now=a[now].ch[CH[i]-'a'];
        for (int t=now;t&&a[t].end!=-1;t=a[t].fail)
        {
            sum+=a[t].end;
            a[t].end=-1;
        }
    }
    return sum;
}
int main()
{
    n=read();
    for (int i=1;i<=n;i++)
    {
        scanf("%s",CH+1);
        build();
    }
    a[0].fail=0;
    get_fail();
    scanf("%s",CH+1);
    printf("%d",ac_query());
    return 0;
}