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

28. 找出字符串中第一个匹配项的下标

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

输入：haystack = "sadbutsad", needle = "sad"
输出：0
解释："sad" 在下标 0 和 6 处匹配。
第一个匹配项的下标是 0 ，所以返回 0 。

思路一：双指针（暴力破解）

class Solution {
    public int strStr(String haystack, String needle) {
        for(int i=0;i<=haystack.length()-needle.length();i++){
            int a=i;
            int b=0;
            while(b<needle.length() && haystack.charAt(a)==needle.charAt(b)){
                a++;
                b++;
            }
            if(b==needle.length()) return i;
        }
        return -1;
    }
}

思路二：KMP算法

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

• 前缀表：next数组就是前缀表(prefix table)

  前缀表是用来回退的，它记录了模式串与主串(文本串)不匹配的时候，模式串应该从哪里开始重新匹配。

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

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

  前缀表的任务是当前位置匹配失败，找到之前已经匹配上的位置，再重新匹配

  前缀表中主要记录下标i之前（包括i）的字符串中，有多大长度的相同前缀后缀

• Next数组(有时候next数组是新前缀表，即旧的前缀表统一减一)

• 时间复杂度：暴力破解的时间复杂度是O(n*m)，而KMP的时间复杂度是O(n+m)

构造next数组的步骤：

1. 初始化
2. 处理前后缀不相同的情况
3. 处理前后缀相同的情况

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

            if(s.charAt(i) == s.charAt(j+1)){
                j++;
            }
            next[i] = j;
        }
    }

完整代码：

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

            if(s.charAt(i) == s.charAt(j+1)){
                j++;
            }
            next[i] = j;
        }
    }
    public int strStr(String haystack, String needle) {
        if(needle.length()==0){
            return 0;
        }

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

        return -1;
    }
}