P3796 【模板】AC自动机(加强版)

P3796 【模板】AC自动机(加强版)

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
#include<queue>
#include<iostream>
using namespace std;
struct node
{
	int ans;
	string data;
	int in;
};
int tree[50000][26],end[50000],fail[50000],tail;
node Ans[50000];
void insert(string a,int num)
{
	int now=0;
	int len=a.length()-1;
	for(int i=0;i<=len;i++)
	{
		int nxt=a[i]-'a';
		if(!tree[now][nxt])
			tree[now][nxt]=++tail;
		now=tree[now][nxt];
	}
	end[now]=num;
}
void get_fail()
{
	queue<int>q;
	for(int i=0;i<26;i++)
		if(tree[0][i])
		{
			q.push(tree[0][i]);
			fail[tree[0][i]]=0;
		}
	while(!q.empty())
	{
		int now=q.front();
		q.pop();
		for(int i=0;i<26;i++)
			if(tree[now][i])
			{
				fail[tree[now][i]]=tree[fail[now]][i];
				q.push(tree[now][i]);
			}
			else
				tree[now][i]=tree[fail[now]][i];
	}
}
void check(string a)
{
	int now=0;
	int len=a.length()-1;
	for(int i=0;i<=len;i++)
	{
		now=tree[now][a[i]-'a'];
		for(int k=now;k;k=fail[k])
			Ans[end[k]].ans+=1;
	}
	return ;
}
bool compare(const node& a,const node& b)
{
	if(a.ans!=b.ans)
		return a.ans>b.ans;
	else	
		return a.in<b.in;
}
int main()
{
	cin.sync_with_stdio(false);
	int n;
	cin>>n;
	while(n)
	{
		tail=0;
		memset(tree,0,sizeof(tree));
		memset(end,0,sizeof(end));
		memset(fail,0,sizeof(fail));
		for(int i=1;i<=n;i++)
		{
			cin>>Ans[i].data;
			insert(Ans[i].data,i);
			Ans[i].ans=0;
			Ans[i].in=i;
		}
		get_fail();
		string end;
		cin>>end;
		check(end);
		sort(Ans+1,Ans+1+n,compare);
		int pas=Ans[1].ans;
		cout<<pas<<endl<<Ans[1].data<<endl;
		for(int i=2;i<=n;i++)
			if(Ans[i].ans==pas)
				cout<<Ans[i].data<<endl;
		cin>>n;
	}
	return 0;
}
posted @ 2018-05-05 15:31  Lance1ot  阅读(167)  评论(0)    收藏  举报