【算法】KMP

KMP算法

场景：根据模式串匹配字符串中模式串的位置

#include <string>
using namespace std;

// KMP算法
int kmp(string s1, string s2){
    // s1 原始字符串，s2模式串
    int n = s1.length(), m = s2.length(),x = 0, y = 0; // s1长度为n，s2长度为m
    // x代表s1中比对的位置， y代表s2中比对的位置
    auto next = nextArray(s2, m);
    while(x < n && y < m) // x y要分别小于原来的长度
    {
        if(s1[x] == s2[y]){ // 两个值相等就都增加
            x++;
            y++;
        }else if(y == 0){ // 两个值不相等，但是y等于0，也就是说模式串的第一个值和当前字符串中的值不等，就比较当前值的下一个
            x++;
        }else{ // 两个值不相等，待比较的模式串位置就跳转到next数组中的对应编号位置
            y = next[y];
        }
    }
    // 如果y等于m证明原字符串中成功匹配到模式串结尾
    // 如果不相等，此时y一定小于m，证明没有成功匹配完模式串，返回-1
    return y == m ? x - y: -1; 
}

// 得到next数组
int* nextArray(string s, int m){
    if(m == 1) {
        return new int[0]{-1};
    }

    int* next = new int[m];
    next[0] = -1;
    next[1] = 0;

    // 0位置代表前面没有前后缀相同的
    // 1位置为也是为0，计算相同前后缀长度不包含本身

    int i = 2, cn = 0;
    while(i < m){
        if(s[i - 1] == s[cn]){
            next[i++] = ++cn;  
        }else if(cn > 0){
            cn = next[cn];  
        }else{
            next[i++] = 0; // cn等于0，代表到next的第一个位置，为-1，则当前的next为0
        }
    }

    return next;
}