C++实现KMP算法

KMP算法原理详解参考链接

KMP算法用C++实现字符串匹配

//根据pattern字符串计算next数组
vector<int> GetNext(const string& p)
{
	vector<int> next(p.size(), -1);
	int k = -1, j = 0;
	while (j < p.size()-1)
	{
		if (k == -1 || p[k] == p[j]){
			k++;
			j++;
			next[j] = k;
		}
		else {
			k = next[k];
		}
	}
	return next;
}

//KMP算法
//计算字符串p在s中是否匹配，匹配时返回匹配的起始位置（从0开始计数），不匹配返回-1
int KMPSearck(const string& s, const string& p)
{
	if (s.empty() || p.empty() || (s.size() < p.size())){
		return -1;
	}
	vector<int> next = GetNext(p);
	int i, j, sLen, pLen;
	i = j = 0;
	sLen = s.size();
	pLen = p.size();
	while (i < sLen && j < pLen)
	{
		if (j == -1 || s[i] == p[j]) {
			i++;
			j++;
		}
		else {
			j = next[j];
		}
	}
	return j == pLen ? i - pLen : -1;
}