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;
}