KMP算法学习历程

1、字符串前缀求法以及KMP变种

https://oi-wiki.org/string/kmp/

KMP 学一遍忘一遍？ACM 金牌选手用可视化直击本质，理解了内核后想忘记都难！_哔哩哔哩_bilibili

文章中pi数组如果只对模式串求就是常规kmp中next数组

稍微变形下就是正常的KMP了

核心还是真前缀的的算法，在失配处利用已经匹配的模式串位置信息直接跳转

2、KMP原论文

阿里云盘分享

3、具体示例代码

#include <iostream>
#include <vector>
#include <string>

using namespace std;

// 计算字符串的前缀函数
vector<int> prefix_function(const string& s) {
    int n = (int)s.length();
    vector<int> pi(n);
    for (int i = 1; i < n; i++) {
        int j = pi[i - 1]; // 从上一个计算过的值开始
        while (j > 0 && s[j] != s[i]) {
            j = pi[j - 1]; // 回溯
        }
        if (s[j] == s[i]) {
            j++;
        }
        pi[i] = j;
    }
    return pi;
}

// KMP模式匹配算法
int KMP(const string& text, const string& pattern) {
    int n = text.length();
    int m = pattern.length();
    vector<int> pi = prefix_function(pattern);
    int i = 0; // text的索引
    int j = 0; // pattern的索引

    while (i < n) {

        if (pattern[j] == text[i]) {
            i++;
            j++;
        }
        if (j == m) {
            return i - j; // 匹配成功，返回匹配的起始下标
        }

        // 如果 目标串没结束 并且 失配
        if (i < n && pattern[j] != text[i]) {
            if (j != 0) {
                j = pi[j - 1]; // 利用前缀函数进行跳转
            }
            else {
                i++;
            }
        }
    }
    return -1; // 未找到匹配
}

// 主函数
int main() {
    string s1 = "ABABDABACDABABCABAB";
    string s2 = "ACDAB";

    // 获取前缀数组
    vector<int> pi = prefix_function(s2);

    // 进行模式匹配
    int matchIndex = KMP(s1, s2);

    // 输出匹配的第一个结果的下标
    if (matchIndex != -1) {
        cout << "Pattern found at index: " << matchIndex << endl;
    }
    else {
        cout << "Pattern not found" << endl;
    }

    cout << endl;

    return 0;
}