模板 转自https://blog.youkuaiyun.com/starstar1992/article/details/54913261
注意 对于cal_next 中 k=next[k] 我的解释:
要比较的是0~k+1 和 p-k~ p 前缀和后缀是否相等(这也是next数组的含义) 若k+1 != p
则 k跳到next[k] 即跳到0~k 的前缀的最后一位(本来k在0~k 的后缀最后一位 )
之后类似递归, 处理 0~k 的前缀, 比较这个和 p-k ~p是否相等(注意此时p-k 不是上一步的k了,已经缩小范围,k变小了,更加靠近p了)
using namespace std;
void cal_next(char *str, int *next, int len)
{
next[0] = -1;//next[0]初始化为-1,-1表示不存在相同的最大前缀和最大后缀
int k = -1;//k初始化为-1
for (int q = 1; q <= len-1; q++)
{
while (k > -1 && str[k + 1] != str[q])//如果下一个不同,那么k就变成next[k],注意next[k]是小于k的,无论k取任何值。
{
k = next[k];//往前回溯
}
if (str[k + 1] == str[q])//如果相同,k++
{
k = k + 1;
}
next[q] = k;//这个是把算的k的值(就是相同的最大前缀和最大后缀长)赋给next[q]
}
}
int KMP(char *str, int slen, char *ptr, int plen)
{
int *next = new int[plen];
cal_next(ptr, next, plen);//计算next数组
int k = -1;
for (int i = 0; i < slen; i++)
{
while (k >-1&& ptr[k + 1] != str[i])//ptr和str不匹配,且k>-1(表示ptr和str有部分匹配)
{
k = next[k];//往前回溯
}
if (ptr[k + 1] == str[i])
k = k + 1;
if (k == plen-1)//说明k移动到ptr的最末端
{
//cout << "在位置" << i-plen+1<< endl;
//k = -1;//重新初始化,寻找下一个
//i = i - plen + 1;//i定位到该位置,外层for循环i++可以继续找下一个(这里默认存在两个匹配字符串可以部分重叠),感谢评论中同学指出错误。
return i-plen+1;//返回相应的位置
}
}
return -1;
}
int main()
{
int next[7];
char *str = "bacbababadababacambabacaddababacasdsd";
char *ptr = "ababaca";
int aa = KMP(str, 36, ptr, 7);
return 0;
}