KMP算法 LeetCode 459. 重复的子字符串

题目

给定一个非空的字符串，判断它是否可以由它的一个子串重复多次构成。给定的字符串只含有小写英文字母，并且长度不超过10000。

示例 1:

输入: "abab"

输出: True

解释: 可由子字符串 "ab" 重复两次构成。
示例 2:

输入: "aba"

输出: False
示例 3:

输入: "abcabcabcabc"

输出: True

解释: 可由子字符串 "abc" 重复四次构成。 (或者子字符串 "abcabc" 重复两次构成。)
           

解题思路

KMP算法

代码

class Solution {
    public boolean repeatedSubstringPattern(String s) {
        return kmp(s + s, s);
    }

    public boolean kmp(String query, String pattern) {
        int n = query.length();
        int m = pattern.length();
        int[] next = new int[m];
        Arrays.fill(next, -1);
        for (int i = 1; i < m; ++i) {
            int j = next[i - 1];
            while (j >=0 && pattern.charAt(j + 1) != pattern.charAt(i)) {
                j = next[j];
            }
            if (pattern.charAt(j + 1) == pattern.charAt(i)) {
                next[i] = j + 1;
            }
        }
        int j = -1;
        for (int i = 1; i < n - 1; ++i) {
            while (j >=0 && pattern.charAt(j + 1) != query.charAt(i)) {
                j = next[j];
            }
            if (pattern.charAt(j + 1) == query.charAt(i)) {
                ++j;
                if (j == m - 1) {
                    return true;
                }
            }
        }
        return false;
        
    }
}