4327: JSOI2012 玄武密码(AC自动机)

4327: JSOI2012 玄武密码

Time Limit: 10 Sec   Memory Limit: 512 MB
Submit: 547   Solved: 242
[ Submit][ Status][ Discuss]

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%的数据,N<=10^7,M<=10^5,每一段文字的长度<=100。


应上传者要求,此题不公开,如有异议,请提出.


Source

[ Submit][ Status][ Discuss]

这题一看数据范围就知道是要用AC自动机了。(其实用后缀数组、hash、奇怪的暴力也可过)

AC之路:


前两次超时是因为我硬套模板,被卡常。其实只用考虑四个字母就够了。

RE嘛……数组开小。。

希望以后不再犯这种低级错误。

讲讲思路吧,只需放在AC自动机上跑一遍,并记录跑过的节点,然后对于每个单词(带有玄武密码的文字)找最深的被跑过的节点,输出即可。

上代码:(这次的码风十分诡异)

#include<bits/stdc++.h>
using namespace std;
const int N = 10000005;
const int C = 4;//注意只有四个,不要开太大,会被卡常。
int idd[N],ch[N][C+1],cnt[N],tot,nxt[N],n,t,ans;
char st[100005][105],stt[N];
int change(char ch){//处理一下
	if(ch=='N') return 0;
	if(ch=='S') return 1;
	if(ch=='W') return 2;
	if(ch=='E') return 3;
}
void insert(int x,int id){//AC自动机(字典树)插入
	int p=0;
	for(int i=0;i<x;i++){
		if(!ch[p][change(st[id][i])]) ch[p][change(st[id][i])]=++tot;
		p=ch[p][change(st[id][i])];
	}
	cnt[p]++;
}
void build(){//建树(失败指针)
	queue<int>q;
	int p=0;
	for(int i=0;i<C;i++){
		if(ch[p][i]) q.push(ch[p][i]);
	}
	while(!q.empty()){
		p=q.front();q.pop();
		for(int i=0;i<C;i++){
			if(ch[p][i]){
				q.push(ch[p][i]);
				nxt[ch[p][i]]=ch[nxt[p]][i];
			}else ch[p][i]=ch[nxt[p]][i];
		}
	}
}
void find(int x){
	int p=0,ret=0;
	for(int i=0;i<x;i++){ 
		p=ch[p][change(stt[i])];
		int k=p;
		while(k&&!idd[k]){
			idd[k]=1;//标记一下
			ret+=cnt[k];
			cnt[k]=0;
			k=nxt[k];
		}
	}
}
void dfs(int x){
	int p=0;
	for(int i=0;i<strlen(st[x]);i++){
		p=ch[p][change(st[x][i])];
		if(idd[p]) ans=i+1;//找标记
	}
}
int main(){
	scanf("%d%d",&t,&n);
 	scanf("%s",stt);
	for(int i=1;i<=n;i++){
			scanf("%s",st[i]);
			insert(strlen(st[i]),i);
		} 
		build();
		find(strlen(stt));
	for(int i=1;i<=n;i++){
			ans=0;
			dfs(i);
			printf("%d\n",ans);
		}
	return 0;
}
perfect++

猜你喜欢

转载自blog.csdn.net/qq_38944163/article/details/81016757