20200612AC自动机(加强版)[模板]

题目描述

给你一个文本串 S 和 n 个模式串 T{1…n}T ，请你分别求出每个模式串 Ti在 S 中出现的次数。

输入格式

第一行包含一个正整数 n 表示模式串的个数。

接下来 n 行，第 i 行包含一个由小写英文字
母构成的字符串 Ti 。

最后一行包含一个由小写英文字母构成的字符
串 S。

输出格式

输出包含 n 行，其中第 ii 行包含一个非负整数
表示T_iT i 在 S 中出现的次数。

输入样例

5
a
bb
aa
abaa
abaaa
abaaabaa

输出样例

6
0
3
2
1

思路：在简单版的基础上将每个模式串打上标记即可。
(即将ed[i]的值改为节点标号，再用ans[i]统计个数求maxans[i])

代码如下：

#include<bits/stdc++.h>
#define N 1001000
using namespace std;
char t[256][256],s[N]
int temp;
int tree[N][26],ed[N],fail[N],last[N],ans[N];
int tot,n;

void Build(char *s,int k)
{
int p=0,L=strlen(s);
for(int i=0;i<L;i++)
{
	int c=s[i]-'a';
	if(!tree[p][c])
	{
		memset(tree[tot],0,sizeof(tree[tot]));
		ed[p]=0;
		tree[p][c]=tot++;
	}
	p=tree[p][c];
}
ed[p]=k;
}

void Getfail()
{
queue<int>q;
int p=0;
fail[0]=0;
for(int i=0;i<26;i++)
{
	p=tree[0][i];
	if(p)
	{
		q.push(p);
		fail[p]=0;
		last[p]=0;
	}
}
while(!q.empty())
{
	int r=q.front();q.pop();
	for(int i=0;i<26;i++)
	{
		p=tree[r][i];
		if(!p)
		{
			tree[r][i]=tree[fail[r]][i];
			continue;
		}
		q.push(p);
		int v=fail[r];
		while(v&&!tree[v][i])v=fail[v];
		fail[p]=tree[v][i];
		if(ed[fail[p]])
		last[p]=fail[p];
		else last[p]=last[fail[p]];
	}
}
}

void Search(char *s)
{
	int p=0,cnt=0,L=strlen(s);
	for(int i=0;i<L;i++)
	{
	int c=s[i]-'a';
	p=tree[p][c];
	for(int j=p;j;j=fail[j])ans[ed[j]]++;
	}
}

int main()
{
while(scanf("%d",&n)&&n!=0)
{
	memset(ans,0,sizeof(ans));
	memset(tree[0],0,sizeof(tree[0]));
	memset(ed,0,sizeof(ed));
	tot=1;
	for(int i=1;i<=n;i++)
	{
		scanf("%s",t[i]);
		Build(t[i],i);
	}
	Getfail();
	scanf("%s",s);
	Search(s);
	temp=0;
	for(int i=1;i<=n;i++)if(ans[i]>temp)temp=ans[i];
	cout<<temp<<endl;
	for(int i=1;i<=n;i++)if(ans[i]==temp)cout<<t[i]<<endl;
}
return 0;
}