​超级实习生计划学习笔记 — Java实现KMP算法匹配字符串的子串​

Java实现KMP算法匹配字符串的子串

public class KMP {
    public static void main(String[] args) throws IOException {
        String s = "abc";
        String s1 = " jhjkhdjabclshgfksdf";
        System.out.println("是否匹配到子串：" + searchStr(s1, s)); 

    }
    public static boolean searchStr(String s1, String s2) {
        //KMP算法
        int[] next = getNext(s2); //获取next数组
        int i = 0;  //i表示s1的下标
        int j = 0;  //j表示s2的下标
        while (i < s1.length() && j < s2.length()) { //当i小于s1的长度且j小于s2的长度时
            if (j == -1 || s1.charAt(i) == s2.charAt(j)) { //当j小于0或者s1的第i个字符与s2的第j个字符相等时
                i++; //i加1
                j++; //j加1
            } else { //当j大于0且s1的第i个字符与s2的第j个字符不相等时
                j = next[j]; //j等于next数组的第j个元素值
            }
        }
        if (j == s2.length()) { //当j等于s2的长度时
            return true; //返回true
        }
        return false;  //否则返回false
    }

    public static int[] getNext(String s) {
        //KMP获取Next
        int[] next = new int[s.length()]; //next数组
        next[0] = -1; //next数组的第一个元素值为-1
        int k = -1; //k表示当前字符串的前缀
        int j = 0; //j表示当前字符串的后缀
        while (j < s.length() - 1) { //当j小于字符串长度-1时
            if (k == -1 || s.charAt(j) == s.charAt(k)) { //当k小于0或者当前字符与前缀字符相等时
                k++; //k加1
                j++; //j加1
                if (s.charAt(j) != s.charAt(k)) { //当当前字符与后缀字符不相等时
                    next[j] = k; //next数组的第j个元素值为k
                } else { //当当前字符与后缀字符相等时
                    next[j] = next[k]; //next数组的第j个元素值为next数组的第k个元素值
                }
            }
            else { //当k大于0且当前字符与前缀字符不相等时
                k = next[k]; //k等于next数组的第k个元素值
            }
        }
        return next;
    }
}