7-1 【模板】KMP字符串匹配

7-1 【模板】KMP字符串匹配
给出两个字符串text和pattern，其中pattern为text的子串，求出pattern在text中所有出现的位置。

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

输入格式:
第一行为一个字符串，即为text。

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

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

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

输入样例:
ABABABC
ABA
输出样例:
1
3
0 0 1
样例说明:

#include<iostream>
#include<bits/stdc++.h>
#include<string.h>

using namespace std;

vector<int> getNext(string pattern)
{
	vector<int> next(pattern.length(), 0);
	int j = 0, k = -1;//j是模式串当前位置，k是最大匹配前缀的下一个位。 
	next[0] = -1;
	while(j < pattern.length()){
		if(k == -1 || pattern[j] == pattern[k]){
			k++;
			j++;
			next[j] = k;
		}else{
			k = next[k];
		}
	}
	return next;
}

void KMP(string text,string pattern)
{
	vector<int> match(text.length(),0);
	vector<int> next = getNext(pattern);
	int j=0;
	int i=0;
	while(i < text.length()){
		if(text[i] == pattern[j]){
			j++;
			i++;
			if(j == pattern.length()){
				cout << i - j + 1 << endl;
				j = next[j];
			}
		}else if(next[j] == -1){
			i++;
		}
		else{
			j = next[j];
		}
	}
}

int main()
{
	string text,pattern;
	cin >> text;
	cin >> pattern ;
	vector<int> next = getNext(pattern);
	KMP(text,pattern);
	for(int i=1;i<=pattern.length();i++)
	{
		if(i!=pattern.length()){
			cout << next[i] << " ";
		}else{
			cout << next[i] << endl;
		}
		
	}
	return 0;
} 

提交之后显示运行时错误，错误原因应该是指针越界。
改进之后的代码：

#include<iostream>
#include<bits/stdc++.h>
#include<string.h>

using namespace std;

vector<int> getNext(string pattern)
{
	vector<int> next(pattern.length()+1,0);
	int  k = 0;//k是最大匹配前缀的下一个位。 
	next[0] = -1;
	next[1] = 0; 
	int j = 1;//j是模式串当前位置 
	while(j < pattern.length()){
		if(k!= -1 && pattern[j] == pattern[k]){
			k++;
			j++;
			next[j] = k;
		}else if(k == -1){
			k = 0;
			j++;
		}else{
			k = next[k];
		}
	}
	return next;
}

void KMP(string text,string pattern)
{
	vector<int> match(text.length(),0);
	vector<int> next = getNext(pattern);
	int j=0;//模式串的下标 
	int i=0;//主串的下标 
	while(i < text.length()){
		if(text[i] == pattern[j]){
			j++;
			i++;
			if(j == pattern.length()){
				cout << i - j + 1 << endl;
				j = next[j];
			}
		}else if(next[j] == -1){
			i++;
		}
		else{
			j = next[j];
		}
	}
}

int main()
{
	string text,pattern;
	cin >> text;
	cin >> pattern;
	vector<int> next = getNext(pattern);
	KMP(text,pattern);
	for(int i=1;i<=pattern.length();i++)
	{
		if(i!=pattern.length()){
			cout << next[i] << " ";
		}else{
			cout << next[i] << endl;
		}	
	}
	return 0;
} 

经过排查，是getNext函数里面的错误，具体原因还不知道，就是改了getNext函数里面的

vector<int> next(pattern.length()+1,0);

在上面加了1，结果就对了。是在创建next容器的时候，分配的空间不足。