KMP算法

KMP算法是求str2是否在str1中存在，然后返回在str1中出现的最早位置
KMP算法中比较关键的一步是求next数组，即位置在i的字符，其前缀与后缀最大相等的个数

package Code_01;

public class KMP {
    public static void main(String[] args) {
        String s1 = "ABCDABDABCDABDABCDABD";
        String s2 = "ABCDABD";
        char[] array2 = s2.toCharArray();
        int[] next = getNextArray(array2);
        for(int n : next){
            System.out.print(n+"  ");
        }
        int a = getIndexOf(s1,s2);
        System.out.println(a);
    }

    public static int getIndexOf(String s1,String s2){
        char[] str1 = s1.toCharArray();
        char[] str2 = s2.toCharArray();
        int[] next = getNextArray(str2);
        int i1 = 0;
        int i2 = 0;
        while(i1 < str1.length && i2 < str2.length){
            if(str1[i1] == str2[i2]){
                i1++;
                i2++;
            }
            else if(next[i2] == -1){
            //说明第一个字符都不相等，就直接str1移动就可以
                i1++;
            }
            else{
            //找到i2的位置，因为前缀等于后缀，不需要从头开始，只需要判断前缀后面的一个字符就好了
                i2=next[i2];
            }
        }
        return i2 == str2.length? i1-i2:-1;
    }
    //求next数组
    public static int[] getNextArray(char[] s){
        int[] next = new int[s.length];
        next[0] = -1;
        next[1] = 0;
        int i = 2;
        int cn = 0;
        //如果前一个数字等于cn所在的位置就+1就可以
        //如果不相等，就要往前找，直到找到某一个位置为0，就可以停止了
        while(i < s.length){
            if(s[i - 1] == s[cn])
            {
                next[i++] = ++cn;
            }
            else if(cn > 0)
            {
                cn = next[cn];
            }
            else{
                    next[i++]=0;
            }
        }
        return next;
    }
}