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;
	}
}