[bzoj4327] 玄武密码

Posted 2021-02-20 hihocoder

Description

在美丽的玄武湖畔，鸡鸣寺边，鸡笼山前，有一块富饶而秀美的土地，人们唤作进香河。相传一日，一缕紫气从天而至，只一瞬间便消失在了进香河中。老人们说，这是玄武神灵将天书藏匿在此。

很多年后，人们终于在进香河地区发现了带有玄武密码的文字。更加神奇的是，这份带有玄武密码的文字，与玄武湖南岸台城的结构有微妙的关联。于是，漫长的破译工作开始了。

经过分析，我们可以用东南西北四个方向来描述台城城砖的摆放，不妨用一个长度为N的序列来描述，序列中的元素分别是‘E’，‘S’，‘W’，‘N’，代表了东南西北四向，我们称之为母串。而神秘的玄武密码是由四象的图案描述而成的M段文字。这里的四象，分别是东之青龙，西之白虎，南之朱雀，北之玄武，对东南西北四向相对应。

现在，考古工作者遇到了一个难题。对于每一段文字，其前缀在母串上的最大匹配长度是多少呢？

Input

第一行有两个整数，N和M，分别表示母串的长度和文字段的个数。

第二行是一个长度为N的字符串，所有字符都满足是E,S,W和N中的一个。

之后M行，每行有一个字符串，描述了一段带有玄武密码的文字。依然满足，所有字符都满足是E,S,W和N中的一个。

Output

输出有M行，对应M段文字。

每一行输出一个数，表示这一段文字的前缀与母串的最大匹配串长度。

Sample Input

7 3
SNNSSNS
NNSS
NNN
WSEE

Sample Output

4
2
0

Hint

对于(100\%)的数据，(Nle 10^7)，(Mle 10^5)，每一段文字的长度(le 100)。

题解

AC自动机，注意一下数据规模

先将所有模式串建好Trie树，然后跑一遍AC自动机，记录某字符串是否是文本串的子串，如果是的话，做好标记

最后一一求模式串的最长前缀即可

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
#include<queue>
#include<stack>
#include<set>
#include<map>
#include<bitset>
#include<vector>
#include<iomanip>
using namespace std;

const int Size=4,Maxn=400001;
struct Trie{
	int son[Maxn][Size],fail[Maxn],flag[Maxn];
	int id;
	int Turn(char c)
	{
		if(c==‘E‘) return 0;
		if(c==‘S‘) return 1;
		if(c==‘W‘) return 2;
		return 3;
	}
	int Newnode()
	{
		for(int i=0;i<Size;++i) son[id][i]=-1;
		flag[id]=0;
		return id++;
	}
	void Init()
	{
		id=0,Newnode();
	}
	void Insert(char s[])
	{
		int l=strlen(s),now=0,k;
		for(int i=0;i<l;++i)
		{
			k=Turn(s[i]);
			if(son[now][k]==-1) son[now][k]=Newnode();
			now=son[now][k];
		}
	}
	queue<int> Q;
	void Build()
	{
		while(!Q.empty()) Q.pop();
		fail[0]=0;
		for(int i=0;i<Size;++i)
			if(son[0][i]==-1) son[0][i]=0;
			else
			{
				fail[son[0][i]]=0;
				Q.push(son[0][i]);
			}
		int now;
		while(!Q.empty())
		{
			now=Q.front(); Q.pop();
			for(int i=0;i<Size;++i)
				if(son[now][i]==-1) son[now][i]=son[fail[now]][i];
				else
				{
					fail[son[now][i]]=son[fail[now]][i];
					Q.push(son[now][i]);
				}
		}
	}
	void Query(char s[])
	{
		int len=strlen(s),now=0;
		for(int i=0;i<len;++i)
		{
			now=son[now][Turn(s[i])];
			int tmp=now;
			while(tmp&&(!flag[tmp]))
			{
				flag[tmp]=1;
				tmp=fail[tmp];
			}
		}
	}
	int Calc(char s[])
	{
		int len=strlen(s),now=0,Ans=0;
		for(int i=0;i<len;++i)
		{
			now=son[now][Turn(s[i])];
			if(flag[now]) Ans=i+1;
			else break;
		}
		return Ans;
	}
}AC;
const int L=10000001;
char s[L],c[100001][120];

int main()
{
	int l,n;
	scanf("%d%d",&l,&n);
	scanf("%s",s);
	AC.Init();
	for(int i=1;i<=n;++i)
	{
		scanf("%s",c[i]);
		AC.Insert(c[i]);
	}
	AC.Build(),	AC.Query(s);
	for(int i=1;i<=n;++i) printf("%d
",AC.Calc(c[i]));
	return 0;
}