PTA-KMP字符串匹配

给出两个字符串text和pattern，其中pattern为text的子串，求出pattern在text中所有出现的位置。

为了减少骗分的情况，接下来还要输出子串的前缀数组next。

输入格式:

第一行为一个字符串，即为text。

第二行为一个字符串，即为pattern。

输出格式:

若干行，每行包含一个整数，表示pattern在text中出现的位置。

接下来1行，包括length(pattern)个整数，表示前缀数组next[i]的值，数据间以一个空格分隔，行尾无多余空格。

输入样例:

ABABABC
ABA

输出样例:

1
3
0 0 1

样例说明:

 AC代码：

#include<bits/stdc++.h>
using namespace std;
void GetNext(string t,int next[]){
	int j=0,k=-1;
	next[0]=-1;
	while(j<t.length()){
		if(k==-1||t[j]==t[k]){
			j++;k++;
			next[j]=k;
		}
		else k=next[k];
	}
}

int main(){
	string s,t;
	cin>>s>>t;
	int i=0;
	int next[100000];
	GetNext(t,next);
	while(s.find(t,i)!=-1){
		cout<<s.find(t,i)+1<<endl;//输出有要求;
		i=s.find(t,i)+1;
	}
	cout<<next[1];
	for(int m=2;m<=t.length();m++){
		cout<<" "<<next[m];
	}
	cout<<endl;
	return 0;
}
//小辣椒的代码

 KMP算法超时代码：欢迎大佬路过，欢迎指正

#include<bits/stdc++.h>
using namespace std;

void GetNext(string t,int next[100000]){
	int j=0,k=-1;
	next[0]=-1;
	while(j<t.length()){
		if(k==-1||t[j]==t[k]){
			j++;k++;
			next[j]=k;
		}
		else k=next[k];
	}
}
int KMP(string s,string t,int i,int next[100000]){
	int n=s.length(),m=t.length();
	int j=0;
	while(i<n&&j<m){
		if(j==-1||s[i]==t[j]){
			i++;
			j++;
		}
		else j=next[j];
	}
	if(j>=m){
		return i-m;
	}
	else return -1;
}
int main(){
	string s,t;
	cin>>s>>t;
	int i=0;
	int next[100000];
	GetNext(t,next);
	while(KMP(s,t,i,next)!=-1){
		cout<<KMP(s,t,i,next)+1<<endl;
		i=KMP(s,t,i,next)+1;
	}
	cout<<next[1];
	for(int m=2;m<=t.length();m++){
		cout<<" "<<next[m];
	}
	cout<<endl;
	return 0;
}