其他分享
首页 > 其他分享> > 【题解】[JSOI2012]玄武密码

【题解】[JSOI2012]玄武密码

作者:互联网

题目链接

题意

输出模式串在文本串中能够匹配上的最长前缀。

输入格式

第一行有两个整数,\(N\)和\(M\),分别表示母串的长度和文字段的个数;

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

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

输出格式

输出有\(M\)行,对应\(M\)段文字。

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

思路

这是一道披上辣椒衣的AC自动机模板题。难点就在这层辣椒衣。

解决方法:先打标记,最后遍历。

代码实现

#include<cstdio>
#include<queue>
#include<cstring>
#include<map>
using namespace std;
const int N=1e5+5;
int n,m;
int t[N*100][5],cnt,fail[N*100];
char s[N][105],txt[N*100];
bool vis[N*100];
map <char,int> mp; 

inline int ans(int x){
	int p=0,ret=0;
	int l=strlen(s[x]);
	for(int i=0;i<l;i++){
		int c=mp[s[x][i]];
		p=t[p][c];
		if(vis[p]) ret=i+1;
	}
	return ret;
}

inline void query(char x[]) {
	int p=0;
	for(int i=0; i<n; i++) {
		p=t[p][mp[x[i]]];
		int j=p;
		while(j && (!vis[j])){
			vis[j]=true;
			j=fail[j];
		}
	}
}

inline void build_graph() {
	queue <int> q;
	for(int i=0; i<4; i++) {
		if(t[0][i]) q.push(t[0][i]);
	}
	while(!q.empty()) {
		int p=q.front();
		q.pop();
		for(int i=0; i<4; i++) {
			if(t[p][i]) {
				q.push(t[p][i]);
				fail[t[p][i]]=t[fail[p]][i];
			} else t[p][i]=t[fail[p]][i];
		}
	}
}

inline void ins(char x[]) {
	int l=strlen(x);
	int p=0;
	for(int i=0; i<l; i++) {
		int c=mp[x[i]];
		if(!t[p][c]) t[p][c]=++cnt;
		p=t[p][c];
	}
}

int main() {
	mp['E']=0; mp['S']=1; mp['W']=2; mp['N']=3;
	scanf("%d%d",&n,&m);
	scanf(" %s",txt);
	for(int i=1; i<=m; i++) {
		scanf(" %s",s[i]);
		ins(s[i]);
	}
	build_graph();
	query(txt);
	for(int i=1; i<=m; i++) printf("%d\n",ans(i));
	return 0;
}

PS. 为了体验不能用万能头的感觉,没有用万能头。

反思:
必须承认,看了题解才写出看上去能算优美的代码。
一开始想要直接把trie的节点弄成包罗万象的大结点,就是说,把每个结点弄成大struct,用每个结点映射它对应的模式串,包括记录它的深度,etc。然后发现写起来很头大。
看了一篇题解,发现可以通过先打标记,最后遍历的方法来确定能匹配的前缀长度。
这是一种存储数据解题(离线)的思维。
事实上,题目提示得很明显:先输入文本串,后输入模式串。这和一般AC自动机输入数据的顺序不同,或许提示着一种更自由地对待模式串的方式(指最后遍历一遍)。

标签:JSOI2012,玄武,题解,结点,int,100,include
来源: https://www.cnblogs.com/Searshkiu/p/16074044.html