左神算法与数据结构——基础提升KMP算法

最新推荐文章于 2024-08-13 09:16:27 发布

冇思想的非菜

最新推荐文章于 2024-08-13 09:16:27 发布

阅读量70

点赞数

分类专栏：基础提升文章标签：数据结构算法 c++

本文链接：https://blog.csdn.net/weixin_53493511/article/details/128446445

版权

基础提升专栏收录该内容

7 篇文章 0 订阅

订阅专栏

KMP算法

解决的问题

在这里插入图片描述

对于短的str2每个k位置下记录之前前缀与后缀相等的最大值

在这里插入图片描述

策略

当str1中i位置与str2中0位置开始比较均相等，直至X、Y位置出现第一次不等。按照图示箭头继续比较。

在这里插入图片描述

由于i1和i1-i2不减，while内语句复杂度为O(2N)

在这里插入图片描述

代码

vector<int> getNexts(string str) {
    vector<int> nexts(str.length());
    if (str.length() == 1) {
        nexts[0] = -1;
        return nexts;
    }
    nexts[0] = -1;
    nexts[1] = 0;
    int cn = 0;
    int i = 2;
    while (i < str.length()) {
        if (str[i - 1] == str[cn]) {
            nexts[i++] = ++cn;
        }
        else if (cn > 0) {
            cn = nexts[cn];
        }
        else {
            nexts[i++] = 0;
        }
    }
    return nexts;
}

int KMP(string str1, string str2) {
    if (str1.length() < 1 || str2.length() < 1 || str1.length() < str2.length()) {
        return -1;
    }
    int i1 = 0;
    int i2 = 0;
    vector<int>nexts = getNexts(str2);
    while (i1 < str1.length() && i2 < str2.length()) {
        if (str1[i1] == str2[i2]) {
            i1++;
            i2++;
        }
        else if (nexts[i2] == -1) {
            i1++;
        }
        else {
            i2 = nexts[i2];
        }
    }
    return i2 == str2.length() ? i1 - i2 : -1;
}