KMP算法_java实现

KMP算法

package com.zhangan.Algorithm.kmp;

import java.util.Arrays;

/**
 * @Author: zhangan
 * @Date: 2021/5/9 11:50
 * @Description: kmp算法
 */
public class KMPAlgorithm {


    /**
     * 获取一个字符串的部分匹配值表
     *
     * @param dest 要匹配的模式串
     * @return 部分匹配值表
     */
    public static int[] kmpNext(String dest) {

        //创建一个next数组保存部分匹配值
        int[] next = new int[dest.length()];
        next[0] = 0;  //如果字符串长度为1  部分匹配值就是0

        for (int i = 1, j = 0; i < dest.length(); i++) {


            //当dest.charAt(i) != dest.charAt(j)，我们需要从next[j-1]获取新的j
            //直到我们发现有dest.charAt(i) == dest.charAt(j)成立才退出
            //KMP的核心语句
            while (j > 0 && dest.charAt(i) != dest.charAt(j)) {
                j = next[j - 1];
            }

            //当dest.charAt(i) == dest.charAt(j) 部分匹配值+1
            if (dest.charAt(i) == dest.charAt(j)) {
                j++;
            }
            next[i] = j;

        }

        return next;
    }

    /**
     * @param str1 原字符串
     * @param str2 要找的字符串
     * @param next 字串的部分匹配表
     * @return 没有->-1  有->返回第一个匹配到的位置
     */
    public static int kemSearch(String str1, String str2, int[] next) {

        for (int i = 0, j = 0; i < str1.length(); i++) {

            //str1.charAt(i) != str2.charAt(j)
            //kmp算法核心
            while (j > 0 && str1.charAt(i) != str2.charAt(j)) {
                j = next[j - 1];
            }


            if (str1.charAt(i) == str2.charAt(j)) {
                j++;
            }

            if (j == str2.length()) {//找到了
                return i - j + 1;
            }

        }

        return -1;

    }

    public static void main(String[] args) {

        String str1 = "BBC ABCDAB ABCDABCDABDE";
        String str2 = "ABCDABD";

        int[] next = kmpNext("ABCDAB");
        System.out.println("部分匹配表" + Arrays.toString(next));
        int index = kemSearch(str1, str2, next);

        if (index == -1) {
            System.out.println("没有该字符串");
        } else {
            System.out.println("该字符串的下标为->" + index);
        }

    }


}