BZOJ 4327 [JSOI2012]玄武密码 (AC自动机)

题目链接: https://www.lydsy.com/JudgeOnline/problem.php?id=4327

题解:

做法挺显然，建出AC自动机之后在上面跑，标记所有走过的点，然后再进行递推，如果(fail[x])被标记则(x)被标记，然后每一个关键点往上找即可

有一个常数优化的技巧: 如果枚举每个(x)再vis[fail[x]]|=vis[x]的话好像还需要按深度排序，不如直接每到一个点直接不停地跳fail一路标记，直到遇到已标记点为止

代码

#include<cstdio>
#include<cstdlib>
#include<cstring>
using namespace std;

const int N = 1e5;
const int LEN = 100;
const int M = 1e7;
const int SIZ = 1e7;
const int S = 4;
char b[N+3][LEN+3];
char a[M+3];
int len[N+3];
int son[SIZ+3][S+3];
int fail[SIZ+3];
int id[SIZ+3];
bool vis[SIZ+3];
int que[SIZ+3];
int ans[N+3];
int n,m,siz;

int decode(char x)
{
	if(x=='E') return 1;
	else if(x=='S') return 2;
	else if(x=='W') return 3;
	else if(x=='N') return 4;
}

void insertstr(int sid)
{
	int u = 0;
	for(int i=1; i<=len[sid]; i++)
	{
		if(son[u][b[sid][i]]==0) {siz++; son[u][b[sid][i]] = siz;}
		u = son[u][b[sid][i]];
	}
	id[u] = sid;
}

void buildACA()
{
	int head = 1,tail = 0;
	for(int i=1; i<=S; i++)
	{
		if(son[0][i]) {tail++; que[tail] = son[0][i];} fail[son[0][i]] = 0;
	}
	while(head<=tail)
	{
		int u = que[head]; head++;
		for(int i=1; i<=S; i++)
		{
			if(son[u][i]) {fail[son[u][i]] = son[fail[u]][i]; tail++; que[tail] = son[u][i];}
			else {son[u][i] = son[fail[u]][i];}
		}
	}
}

int main()
{
	scanf("%d%d",&m,&n);
	scanf("%s",a+1); for(int i=1; i<=m; i++) a[i] = decode(a[i]);
	for(int i=1; i<=n; i++)
	{
		scanf("%s",b[i]+1); len[i] = strlen(b[i]+1);
		for(int j=1; j<=len[i]; j++) b[i][j] = decode(b[i][j]);
		insertstr(i);
	}
	buildACA();
	int u = 0; vis[0] = true;
	for(int i=1; i<=m; i++)
	{
		vis[u] = true;
		u = son[u][a[i]];
		for(int j=u; vis[j]==false; j=fail[j]) {vis[j] = true;}
	}
	for(int i=1; i<=n; i++)
	{
		u = 0;
		for(int j=1; j<=len[i]; j++)
		{
			u = son[u][b[i][j]];
			if(!vis[u]) {break;}
			ans[i]++;
		}
	}
	for(int i=1; i<=n; i++) printf("%d
",ans[i]);
	return 0;
}