Java数据结构之字符串模式匹配算法---KMP算法2

直接接上篇上代码：

 //KMP算法
 public class KMP {

     // 获取next数组的方法,根据给定的字符串求
     public static int[] getNext(String sub) {

         int j = 1, k = 0;
         int[] next = new int[sub.length()];
         next[0] = -1; // 这个是规定
         next[1] = 0; // 这个也是规定
         //
         while (j < sub.length() - 1) {
             if (sub.charAt(j) == sub.charAt(k)) {
                 next[j + 1] = k + 1;
                 j++;
                 k++;
             } else if (k == 0) {
                 next[j + 1] = 0;
                 j++;
             } else {
                 k = next[k];
             }

         }
         return next;
     }

     // 根据给定的主串和子串，采用KMP算法来获取模式匹配
     public static int kmp(String src, String sub) {

         // 首先生成模式串sub的next[j]
         int[] next = getNext(sub);
         int i = 0, j = 0, index = -1;
         while (i < src.length() && j < sub.length()) {
             if (src.charAt(i) == sub.charAt(j)) {
                 i++;
                 j++;
             } else if (j == 0) {
                 i++;
             } else {
                 j = next[j];
             }
         }

         // 得到开始匹配的位置索引
         if (j == sub.length()) {
             index = i - sub.length();
         }
         return index;
     }
 }

//

 //KMP算法的测试
 public class Test {
     /**
      * @param args
      */
     public static void main(String[] args) {
         String src = "aaaaaaab";
         String sub = "ABCDABDABD";
         int[] next = KMP.getNext(sub);
         //int[] next = lengthKMP(sub.toCharArray());
         for (int i : next) {
             System.out.print(i + "  ");// -1 0 1 2 3
         }

         // System.out.println();
         // System.out.println(KMP.kmp(src, sub));
     }

     //补充上一篇中的对于前缀后缀的讨论的获取部分匹配数组的算法
     public static int[] lengthKMP(char[] mchar) {
         int[] fixNum = new int[mchar.length];
         for (int i = 1, j = 0; i < mchar.length; i++) {
             if (mchar[j] == mchar[i]) {
                 fixNum[i] = j + 1;
                 j++;
             } else if (j > 0) {
                 j = 0;
                 i -= j;
             }
         }
         // return [0, 0, 0, 0, 1, 2, 0, 1, 2, 0]ABCDABDABD
         return fixNum;
     }
 }