KMP算法进行字符串匹配

示例代码如下

public class KMPAlgorithm {
    public static void main(String[] args) {
        String srcStr = "BBC ABCDAB ABCDABCDABDE";
        String dest = "ABCDABD";

        int[] matchTable = getPartMatchTable(dest);
        int index = searchIndex(srcStr, dest, matchTable);
        System.out.println(index);
    }

    /**
     * @param src        源字符串
     * @param dest       子串
     * @param matchTable 子串的部分匹配表
     * @return 如果匹配上了返回子串在源字符串中的起始下标, 否则返回-1
     */
    public static int searchIndex(String src, String dest, int[] matchTable) {
        int length = src.length();
        int destLength = dest.length();
        for (int i = 0, j = 0; i < length; i++) {
            while (j > 0 && src.charAt(i) != dest.charAt(j)) {
                // 如果当前字符不相等,则根据部分匹配表重新给j赋值
                j = matchTable[j - 1];
            }
            if (src.charAt(i) == dest.charAt(j)) {
                // 若当前字符相等就继续比较下一个字符
                j++;
            }
            if (j == destLength) {
                return i - j + 1;
            }
        }
        return -1;
    }

    /**
     * 获取子串的部分匹配表
     *
     * @param dest
     * @return
     */
    public static int[] getPartMatchTable(String dest) {
        int length = dest.length();
        int[] rst = new int[length];
        for (int i = 1, j = 0; i < length; i++) {
            while (j > 0 && dest.charAt(i) != dest.charAt(j)) {
                j = rst[j - 1];
            }
            if (dest.charAt(i) == dest.charAt(j)) {
                j++;
            }
            rst[i] = j;
        }
        return rst;
    }
}