KMP算法模板 - 构建next最长前缀数组 与 kmp核心算法

#include <iostream>
#include <string>
using namespace std;
//构建next最长前缀数组
int* getNextArray(const string &sub)
{
	if(sub.length() == 1) {//只有一个字符
		return new int[1] {-1};
	}
	int* next = new int[sub.size()];
	next[0] = -1;
	next[1] = 0;
	int i = 2;
	int cn = 0;//用于保存前一个字符的相同前缀的长度
	while (i < sub.size()){
		if(sub[i - 1] == sub[cn]){//若又匹配的上，则再进行++，继续往后一个字符进行判断
			++cn;
			next[i] = cn;
			++i;
		}else if(cn > 0){//若匹配不上，则将长度缩短为-原来匹配长度的位置-的 相同前缀 的长度
			cn = next[cn];
		}else{//若cn = 0，即完全没有相同前缀的了
			next[i] = 0;
			++i;
		}
	}
	return next;
}
//kmp匹配算法
int getIndexOf(const string &main, const  string &sub)
{
	if(main.empty() || sub.empty() || main.length() < sub.length()){
		return -1;
	}
	int i1 = 0, i2 = 0;
	int* next = getNextArray(sub);
	while (i1 < main.length() && i2 < sub.length())
	{
		if(main[i1] == sub[i2])//正常匹配
		{
			++i1;
			++i2;
		}else if(next[i2] == -1)//子串第一个数就不匹配
		{
			++i1;
		}else//匹配不上
		{
			i2 = next[i2];
		}
	}
	return i2 == sub.size() ? i1 - i2 : -1;//完全匹配 or 匹配失败
}
int main()
{
	const string main = "123456789abacdcabaegbfios", sub = "abacdcabae";
	cout << getIndexOf(main, sub) << endl;// 9
	return 0;
}

题外话：

可以直接使用string.find("")来获取答案