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

KMP

KMP的主要思想是当出现字符串不匹配时，可以知道一部分之前已经匹配的文本内容，可以利用这些信息避免从头再去做匹配了。

KMP的精髓所在就是前缀表，在KMP精讲 (opens new window)中提到了，什么是KMP，什么是前缀表，以及为什么要用前缀表。

前缀表：起始位置到下标i之前（包括i）的子串中，有多大长度的相同前缀后缀。

那么使用KMP可以解决两类经典问题：

1. 匹配问题：28. 实现 strStr()(opens new window)
2. 重复子串问题：459.重复的子字符串(opens new window)

前缀：指不包含最后一个字符的所有以第一个字符开头的连续子串。

后缀：指不包含第一个字符的所有以最后一个字符结尾的连续子串。

然后针对前缀表到底要不要减一，这其实是不同KMP实现的方式，我们在KMP精讲 (opens new window)中针对之前两个问题，分别给出了两个不同版本的的KMP实现。

其中主要理解j=next[x]这一步最为关键！

28. 实现 strStr()

力扣题目链接(opens new window)

实现 strStr() 函数。

给定一个 haystack 字符串和一个 needle 字符串，在 haystack 字符串中找出 needle 字符串出现的第一个位置 (从0开始)。如果不存在，则返回  -1。

示例 1: 输入: haystack = "hello", needle = "ll" 输出: 2

示例 2: 输入: haystack = "aaaaa", needle = "bba" 输出: -1

说明: 当 needle 是空字符串时，我们应当返回什么值呢？这是一个在面试中很好的问题。 对于本题而言，当 needle 是空字符串时我们应当返回 0 。这与C语言的 strstr() 以及 Java的 indexOf() 定义相符。

思路

KMP（Knuth-Morris-Pratt）算法是一种用于字符串匹配的高效算法，其主要目的是在文本串中查找给定的模式串。下面是KMP算法的流程：

1. 预处理模式串：

   • 构建一个辅助数组next，用于存储模式串中每个位置的最长公共前后缀的长度。
   • 初始化两个指针i和j，i指向模式串的第一个字符，j指向模式串的第二个字符。
   • 设置next[0]为-1，next[1]为0。
   • 进入循环，遍历模式串的每个字符：
     • 若模式串中i位置的字符等于j位置的字符，则将next[i+1]设为j+1，i和j都增加1。
     • 若模式串中i位置的字符不等于j位置的字符：
       • 若j等于0，则将next[i+1]设为0，i增加1。
       • 若j不等于0，则将j更新为next[j]，继续进行比较。

2. 在文本串中查找模式串：

   • 初始化两个指针i和j，i指向文本串的第一个字符，j指向模式串的第一个字符。
   • 进入循环，遍历文本串的每个字符：
     • 若文本串中i位置的字符等于模式串中j位置的字符：
       • 若j等于模式串的长度减一，则说明找到了完整的匹配，返回匹配位置i-j。
       • 否则，i和j都增加1。
     • 若文本串中i位置的字符不等于模式串中j位置的字符：
       • 若j等于0，则将i增加1。
       • 若j不等于0，则将j更新为next[j]。

3. 若循环结束后仍未找到匹配，则返回匹配失败。

KMP算法通过利用模式串的最长公共前后缀信息，在匹配过程中尽可能地减少不必要的比较次数，提高了匹配的效率。这种预处理的方式使得KMP算法的时间复杂度为O(m+n)，其中m为模式串的长度，n为文本串的长度。

class Solution {
    //前缀表（不减一）Java实现
    public int strStr(String haystack, String needle) {
        if (needle.length() == 0) return 0;
        int[] next = new int[needle.length()];
        getNext(next, needle);

        int j = 0;
        for (int i = 0; i < haystack.length(); i++) {
            while (j > 0 && needle.charAt(j) != haystack.charAt(i)) 
                j = next[j - 1];
            if (needle.charAt(j) == haystack.charAt(i)) 
                j++;
            if (j == needle.length()) 
                return i - needle.length() + 1;
        }
        return -1;

    }
    
    private void getNext(int[] next, String s) {
        int j = 0;
        next[0] = 0;
        for (int i = 1; i < s.length(); i++) {
            while (j > 0 && s.charAt(j) != s.charAt(i)) 
                j = next[j - 1];
            if (s.charAt(j) == s.charAt(i)) 
                j++;
            next[i] = j; 
        }
    }
}