【洛谷P3796】【模板】AC自动机（加强版）

题目描述

有个由小写字母组成的模式串以及一个文本串。每个模式串可能会在文本串中出现多次。你需要找出哪些模式串在文本串中出现的次数最多。

输入输出格式

输入格式：
输入含多组数据。

每组数据的第一行为一个正整数，表示共有个模式串，。

接下去行，每行一个长度小于等于的模式串。下一行是一个长度小于等于的文本串。

输入结束标志为。

输出格式：
对于每组数据，第一行输出模式串最多出现的次数，接下去若干行每行输出一个出现次数最多的模式串，按输入顺序排列。

输入输出样例

输入样例#1：
2
aba
bab
ababababac
6
beta
alpha
haha
delta
dede
tata
dedeltalphahahahototatalpha
0
输出样例#1：
4
aba
2
alpha
haha

题解
在简单版上稍加改动，记录每个单词结束的节点位置，记录出现次数即可

代码

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
char s[151][71],m[1000001];
int tot,n,a[1000001][26],danger[1000001],point[1000001],Hash[1000001],ans[151];
int q[1000001];
inline void ins(int x)
{
    int now=1,l=strlen(s[x]);
    for (int i=0;i<l;i++)
    {
        int t=s[x][i]-'a';
        if (!a[now][t]) a[now][t]=++tot;
        now=a[now][t];
    }
    Hash[now]=x;
}
inline void acmach()
{
    int Head=0,tail=1;
    q[1]=1;point[1]=0;
    while (Head!=tail)
    {
        int now=q[++Head];
        for (int i=0;i<26;i++)
            if (a[now][i])
            {
                point[a[now][i]]=a[point[now]][i];
                q[++tail]=a[now][i];
            }
            else a[now][i]=a[point[now]][i];
    }
}
inline void solve()
{
    int now=1,l=strlen(m);
    for (int i=0;i<l;i++)
    {
        int t=m[i]-'a';
        now=a[now][t];
        for (int j=now;j;j=point[j])
            if (Hash[j]) ans[Hash[j]]++;
    }
}
int main()
{
    n=read();
    for (int i=0;i<26;i++) a[0][i]=1;
    while (n)
    {
        tot=1;
        for (int i=1;i<=n;i++)
        {
            scanf("%s",s[i]);
            ins(i);
        }
        acmach();
        scanf("%s",m);
        solve();
        int mx=0;
        for (int i=1;i<=n;i++)
            mx=max(mx,ans[i]);
        printf("%d\n",mx);
        for (int i=1;i<=n;i++)
        {
            if (mx==ans[i]) puts(s[i]);
            ans[i]=0;
        }
        for (int i=1;i<=tot;i++)
        {
            Hash[i]=point[i]=danger[i]=0;
            for (int j=0;j<26;j++)
                a[i][j]=0;
        }
        n=read();
    }
    return 0;
}