KMP算法

kmp算法

package com.ming;

import java.util.ArrayList;


public class Code4 {
    public static void main(String[] args) {
        String s1 = "abc";
        String s2= "abca";//
        System.out.println(KMP(s1,s2));


    }
    public static int KMP(String s1,String s2){//主函数
        if(s1 == null || s1.length() == 0 || s2==null || s2.length() == 0) return -1;
        int i = 0;//主字符串串索引
        int j = 0;//子字符串索引
        int[] next = getArr(s1);//获取next数组
        while (i<s1.length()&&j<s2.length()){//任何一个越界便退出循环，
            if(s1.charAt(i) == s2.charAt(j)){//当字符相同时，两个索引同时++
                i++;j++;
            } else if(next[j]==-1){//等价于j==0，若是j向前回跳到0则表明子和主字符串在j和i位置处字符不相等，主字符串应该右移一位
                i++;
            }else {//若是上一个if未成立，则找到next数组中需要回退的下一个位置
                j = next[j];
            }
        }
		//返回对应位置的索引，当i或j越界时，判断j是否等于子字符串的长度，若是则返回j-i，否则返回-1
        return j ==s2.length() ? j-i:-1;
    }


    public static int[] getArr(String s1){
        if (s1.length() == 1) {//子字符串的长度为1，即没有相同的最长真前后子缀，设为-1
            return new int[] {-1};
        }
        char[] chars = s1.toCharArray();
        int[] next = new int[s1.length()];
        next[0]=-1;
        next[1]=0;
        int i=2;
        int cn = 0;
        while(i<next.length){
            if(chars[cn]==chars[i-1]){
                next[i++]=++cn;
            }else if(cn>0){
                cn = next[cn];
            }else{
                next[i++] = 0;
            }
        }
        return next;
    }
}