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容器的时候,分配的空间不足。