KMP算法的Java实现

输入

第一行一个整数N，表示测试数据组数。

接下来的N*2行，每两行表示一个测试数据。在每一个测试数据中，第一行为模式串，由不超过10^4个大写字母组成，第二行为原串，由不超过10^6个大写字母组成。

其中N<=20

输出

对于每一个测试数据，按照它们在输入中出现的顺序输出一行Ans，表示模式串在原串中出现的次数。

import java.util.Scanner;public class Main{public static void main(String[] args) {Scanner sc = new Scanner(System.in);int N = Integer.parseInt(sc.nextLine());String pattern = "", str = "";for(int i = 0; i < N; i++){pattern = sc.nextLine();str = sc.nextLine();System.out.println(KMP(pattern, str));}}public static int KMP(String pattern, String str) {int[] next = calNext(pattern);int count = 0, i = 0, j = 0;//初始状态while(i < str.length()){if(str.charAt(i) == pattern.charAt(j)){//当前字符相同if(j == pattern.length() - 1){//匹配成功count++;i++;j = next[j];}else{//继续匹配下一字符i++;j++;}}else{//当前字符不同if(j == 0){//如果已经是模式串的第一个字符i++;}else{j = next[j - 1];}}}return count;}/** * 计算next数组 * @param pattern 模式串 * @return */public static int[] calNext(String pattern) {int[] next = new int[pattern.length()];next[0] = 0;int i = 1, j = 0;while(i < pattern.length()){if(pattern.charAt(i) == pattern.charAt(j)){next[i++] = ++j;}else{if(j == 0){next[i++] = 0;}else{j = next[j-1];}}}return next;}}