AC自动机->RE自动机?
查看原帖
AC自动机->RE自动机?
305891
夜明楼主2021/6/6 15:14

50,第2个数据点怎么调都是 RE 。

求助大佬!

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
struct node{
	int nxt[26]={0};
	int ans=0;
}son[1000005];
int n,tot=1,que[1000005],nxt[1000005];
char mainn[1000005],pattern[1000005];
void Build(char *s){
	int len=strlen(s+1),cnt=1;
	for(int i=1;i<=len;i++){
		int c=s[i]-'a';
		if(!son[cnt].nxt[c])
			son[cnt].nxt[c]=++tot;
		cnt=son[cnt].nxt[c];
	}
	son[cnt].ans++;
} 
void Pre(int x,int y){
	int z=son[x].nxt[y],last=nxt[x];
	nxt[z]=son[last].nxt[y];
}
void BFS(){
	for(int i=0;i<26;i++)
		son[0].nxt[i]=1;
	que[1]=1;
	int l=1,r=1;
	while(l<=r){
		int x=que[l];
		l++;
		for(int i=0;i<26;i++)
			if(son[x].nxt[i]){
				que[++r]=son[x].nxt[i];
				Pre(x,i);
			}else
				son[x].nxt[i]=son[nxt[x]].nxt[i];
	}
}
int KMP(char *s){
	int len=strlen(s+1),cnt=1,sum=0;
	for(int i=1;i<=len;i++){
		int c=s[i]-'a';
		int k=son[cnt].nxt[c];
		cnt=k;
		while(k>1){
			sum+=son[k].ans;
			son[k].ans=0;
			k=nxt[k];
		}
	}
	return sum;
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		scanf("%s",pattern+1);
		Build(pattern);
	}
	BFS();
	scanf("%s",mainn+1);
	printf("%d",KMP(mainn));
	return 0;
} 
2021/6/6 15:14
加载中...