代码随想录算法训练营第九天|28.KMP算法

题目

题目描述

题目链接：力扣28.找出字符串中第一个匹配的下标

给你两个字符串 haystack 和 needle ，请你在 haystack 字符串中找出 needle 字符串的第一个匹配项的下标（下标从 0 开始）。如果 needle 不是 haystack 的一部分，则返回 -1 。

思路

KMP算法的核心是next数组，这里只记录如何生成next数组。

其实，求next数组的过程完全可以看成字符串匹配的过程，即以模式字符串为主字符串，以模式字符串的前缀为目标字符串，一旦字符串匹配成功，那么当前的next值就是匹配成功的字符串的长度。

具体来说，就是从模式字符串的第一位(注意，不包括第0位)开始对自身进行匹配运算。 在任一位置，能匹配的最长长度就是当前位置的next值。

参考文章

1. 知乎@海纳的回答：如何更好地理解和掌握 KMP 算法?
2. 力扣@宫水三叶题解：【宫水三叶】简单题学 KMP 算法
3. 代码随想录讲解：代码随想录28.实现strStr()

代码实现

class Solution {
public:
    int strStr(string haystack, string needle) {
        int next[10000];
        next[0] = -1;
        int i = 0, j = -1;
        while(i < needle.size()){
            if(j == -1 || needle[i] == needle[j]){
                i++;
                j++;
                next[i] = j;
            }
            else{
                j = next[j];
            }
        }
        
        i = 0;
        j = 0;
        while(i < haystack.size() && j < needle.size()){
            if(j == -1 || haystack[i] == needle[j]){
                i++;
                j++;
            }
            else{
                j = next[j];
            }
        }
        if(j == needle.size()) return i - j;
        
        return -1;
    }
};