代码随想录算法训练营09|kmp算法

今天的两道题目都跟kmp算法有关，对于kmp算法的原理和思想，个人认为已经掌握。在看完carl对kmp算法的讲解后，存在着一个关键的疑问导致一直无法敲出源代码：如何把手算转化为代码层面？next数组初始化中j的回退规则为什么是j=next[j]？对于这个疑问，在b站看到这个视频后迎刃而解。

那么对于理解了kmp算法后，根据思路进行做题就基本ok了。

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

class Solution {
    public int strStr(String haystack, String needle) {
        int length=needle.length();
        int[] Next=new int[needle.length()];
        Next[0]=0;
        getNext(Next,needle);
        int j=0;
        for(int i=0;i<haystack.length();i++){
            while(j>0&&haystack.charAt(i)!=needle.charAt(j)){
                j=Next[j-1];
            }
            if(haystack.charAt(i)==needle.charAt(j)){
                j++;
            }
            if(j==needle.length())
                return i-needle.length()+1;
        }
        return -1;
    }
    public void getNext(int[] Next,String needdle){
        int j=0;
        for(int i=1;i<needdle.length()-1;i++){
            while(j>0&&needdle.charAt(i)!=needdle.charAt(j)){
                j=Next[j-1];
            }
            if(needdle.charAt(i)==needdle.charAt(j)){
                Next[i]=++j;
            }
        }
    }
}

459. 重复的子字符串

老实说，因为这题是在kmp的章节里面，所以我直接就往kmp算法的思路上靠了。

我的第一想法是，如果这个字符串由某个字符串重复多次构成，那么从第二个字串开头开始next数组随着下标增1也跟着增1，因此我的最初判断条件是在完成next数组的构造后，直接进行next数组遍历，从遇到的第一个1开始如果next[i]==next[i-1]+1一直满足到遍历结束则返回正确。

但是，很遗憾的是解答错误，问题就在于，无法知道这个重复的子串到底何时结束。我最初的想法是第一个重复子串的next数组都是0（很美好是不，确实在运行给出的两个例子ac了），但是关键是这个重复子串中不可避免地会存在着相等的前后缀，因此他的next数组不一定全为0.

这个方法失败后，我又陷入了思考，现在我能确定是，如果这个字符串真的是重复子串构成，那么这个重复次数至少是2，那么next数组的最大next[j]一定是在末尾而且是大于len/2。

ok那这能不能作为判断条件呢，经过debug后还是不行，为什么？因为例如ababa，同样这个字符串中，最长相等前后缀是3超过了len/2的2，但是中间的“a”是被两个子串共享的，并不符合重复子串的这个概念。当发现这个错误后，其实跟答案就只相差一步，那就是len%（len-next[len]）==0！

class Solution {
    public boolean repeatedSubstringPattern(String s) {
        if (s.equals("")) return false;

        int len = s.length();
        // 原串加个空格(哨兵)，使下标从1开始，这样j从0开始，也不用初始化了
        s = " " + s;
        char[] chars = s.toCharArray();
        int[] next = new int[len + 1];

        // 构造 next 数组过程，j从0开始(空格)，i从2开始
        for (int i = 2, j = 0; i <= len; i++) {
            // 匹配不成功，j回到前一位置 next 数组所对应的值
            while (j > 0 && chars[i] != chars[j + 1]) j = next[j];
            // 匹配成功，j往后移
            if (chars[i] == chars[j + 1]) j++;
            // 更新 next 数组的值
            next[i] = j;
        }

        // 最后判断是否是重复的子字符串，这里 next[len] 即代表next数组末尾的值
        if (next[len] > 0 && len % (len - next[len]) == 0) {
            return true;
        }
        return false;
    }
}