KMP算法的JAVA实现

什么是KMP算法

Knuth-Morris-Pratt算法（简称KMP）是经常使用的字符串匹配算法之一。

假设如今有一个模式串a="ABACABAD"和一个主串b="BBC ABACABACABAD ABCDABDE"，要判断主串b是否包含模式串a，若是包含，则返回出模式串在主串的位置下标。

易知使用暴力匹配算法的时间复杂度为O(m*n),其中m和n为模式串和主串的长度。而使用KMP算法，则能在线性时间O(m+n)中完成匹配工做。

KMP算法实现逻辑

使用暴力匹配算法时，每次不匹配，都须要从主串下一个位置从头匹配一次模式串，这种回溯工做，致使效率低下。KMP算法核心思想是充分利用上次不匹配时的计算结果，避免"一切从新开始"的计算工做。

如下经过一个简单的例子进行说明：

一、首先，使用主串的第一位与模式串的第一位进行比较，若是不一样，则将主串的第二位与模式串的第一位进行比较，以此类推。

比较主串第一位与模式串第一位的字符比较主串第二位与模式串第一位的字符

二、直到主串有一个字符与模式串的第一位相同，则比较主串下一个位置的字符，是否与模式串的第二位相同，以此类推。

主串中的字符匹配到模式串的第一位字符比较主串的下一个字符与模式串的第二位字符

三、当匹配到某个位置，主串与模式串的字符不一样时，此时不直接从主串下一个位置，再从头逐个比较。由于在比较过程当中，咱们能够知道两个细节：

（1）模式串的前面部分的字符串内容是与主串的部分字符是相同的。

（2）在该模式串"ABACABAD"中，下标0~2的字符是与下标4~6的字符是相同的。

所以，咱们直接使用下标位置为3的字符与主串进行比较，这样就能大大提升效率了。

主串字符C与模式串D不匹配模式串下标0~2的字符是与下标4~6的字符相同，所以也与主串的前三个位置的字符是匹配的不重头开始比较，而是比较模式串下标3的字符与主串中的字符是否相同

四、以此类推，直到匹配到模式串的最后一位，或者扫描完主串。

匹配到模式串的最后一位

部分匹配表

在匹配步骤3中，其实利用了模式串自己字符的组合顺序信息，在KMP算法中，咱们须要将该字符组合顺序信息记录起来，称之为"部分匹配表"。

"部分匹配表"是如何产生的呢？首先，要了解两个概念："前缀"和"后缀"。 "前缀"指除了最后一个字符，一个字符串的所有头部组合,"后缀"指除了第一个字符，一个字符串的所有尾部组合。

例如字符串a="ABCAB"，前缀字符串集合为[A, AB, ABC,ABCA]，后缀字符串集合为[B, AB, CAB,BCAB]，能够看到前缀和后缀有相同的子串[AB]。

部分匹配值，其实就是计算出下标在0~i的子字符串中(i<=a.length)，前缀与后缀最长相同子串的长度。

"部分匹配表"计算规则可参考阮一峰老师的日志“ 字符串匹配的KMP算法”。

咱们根据这个规则，可计算模式串a="ABACABAD"的部分匹配表，以下：

KMP算法的JAVA 代码实现

1.计算部分匹配值。

public static int[] kmpnext(String dest){
    int[] next = new int[dest.length()];
    next[0] = 0;

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

代码说明：

1）声明部分匹配表数组，用于存储匹配值。

2）当字符串为空字符串str="",没有先后缀字符串，所以最长匹配值为0，next[0] = 0。

3）循环字符串，计算出下标在0~i的字符串的部分匹配表，i初始化为1。j用于记录前缀与后缀最长相同子串的长度。

(a) 若是在0~i的子字符串，j=0，而且dest.charAt(j) != dest.charAt(i)时，表示在0~i这一段中，先后缀字符串集合中没有相同字符串，所以next[i]=j（即next[i]=0）。

(b) 若是在0~i的子字符串，j=0，dest.charAt(j) == dest.charAt(i)时，表示在0~i这一段中，先后缀字符串集合中有一个字符串相同，所以j++;next[i]=j;（即next[i]=1）。

(c) 若是在0~i的子字符串，dest.charAt(j) == dest.charAt(i)时，若是j>0，则表示上一轮比较，在0~i-1的子字符串中，前缀与后缀有相同子串。所以在0~i这一段中，前缀与后缀也有相同子串，而且最长的共有字符串长度为j++。所以j++;next[i]=j。

(d) 若是在0~i的子字符串，j>0，dest.charAt(j) != dest.charAt(i)时，则表示上一轮比较时，字符串[0~j-1]是字符串[0~i-1]中，先后缀的最长相同字符串，若是咱们找到在字符串[0~j-1]中的最长先后缀相同字符串(记做maxComStr)，继续比较maxComStr下一位与dest.charAt(i)，则能减小比较次数。经过部分匹配表中可见，next[j-1]为[0~j-1]中先后缀最长相同字符串的长度，咱们也能够理解为是最长相同字符串下一个字符的下标，所以j=next[j-1]，举例说明：

dest.charAt(j) != dest.charAt(i)字符串[0~j-1]中，前缀字符串集合为[A,AB]，后缀字符串集合为[A,BA],最长共有元素为A,j=next[j-1],则j移动到了该最长前缀字符串下一位继续比较该最长前缀字符串下一位与dest.charAt(i)

2.比较模式串和主串。

public static int kmp(String str, String dest){
    //1.首先计算出部分匹配表
    int[] next = kmpnext(dest);
    //2.查找匹配位置
    for(int i = 0, j = 0; i < str.length(); i++){
        while(j > 0 && str.charAt(i) != dest.charAt(j)){
            j = next[j-1];
        }
        if(str.charAt(i) == dest.charAt(j)){
            j++;
        }
        if(j == dest.length()){
            return i-j+1;
        }
    }
    return -1;
}

代码说明：

1）计算部分匹配表。

2）j为模式串a下标，i为主串b下标。循环主串，查找匹配位置。

(1) 若是j=0，而且str.charAt(i) != dest.charAt(j)时，则移动主串下标位置，比较主串下一位字符是否与模式串第一位字符相同。

(2) 若是str.charAt(i) == dest.charAt(j)时，则同时移动主串下标位置和模式串下标位置，依次比较下一位。

(3) 若是比较到模式串某个位置(j>0)，str.charAt(i) != dest.charAt(j)时，则根据部分匹配表，移动到[0~j-1]字符串的先后缀最长相同字符串的后一位，继续进行比较。如在该模式串dest ="ABACABAD"中，当j=7时，dest.charAt(7)与主串的字符不一样。而dest[0~6]这部分字符串是与主串str[i-6~i-1]匹配的，dest[0~2]字符是与dest[4~6]的字符是相同的，由此能够推断出dest[0~2]的字符也与主串str[i-3~i-1]的字符是相同的。经过部分匹配表中可见，next[j-1]为先后缀最长相同字符串的长度，咱们也能够理解为是最长相同字符串下一个字符的下标，所以j=next[j-1]。

(4) 当j == dest.length()时，代表完成模式串的比较，返回匹配起始位置(i-j+1)。

完整代码以下：

public class Kmp {

    public static void main(String[] args){
        String a = "ABACABAD";
        String b = "BBC ABACABACABAD ABCDABDE";
        int result = kmp(b, a);

        //打印结果：和字符串得到匹配的位置        System.out.println("resultPosion:"+result);
    }

    /**     * KMP 匹配     */
    public static int kmp(String str, String dest){
        //1.首先计算出 部分匹配表        int[] next = kmpnext(dest);

        System.out.println("next ="+Arrays.toString(next));
        //2.查找匹配位置        for(int i = 0, j = 0; i < str.length(); i++){
            while(j > 0 && str.charAt(i) != dest.charAt(j)){
                j = next[j-1];
            }
            if(str.charAt(i) == dest.charAt(j)){
                j++;
            }
            if(j == dest.length()){
                return i-j+1;
            }
        }
        return -1;
    }

    /**     * 计算部分匹配表     */
    public static int[] kmpnext(String dest){
        int[] next = new int[dest.length()];
        next[0] = 0;

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