hihoCoder:KMP

题目1 : KMP算法

时间限制:1000ms
单点时限:1000ms
内存限制:256MB

描述

小Hi和小Ho是一对好朋友，出生在信息化社会的他们对编程产生了莫大的兴趣，他们约定好互相帮助，在编程的学习道路上一同前进。
这一天，他们遇到了一只河蟹，于是河蟹就向小Hi和小Ho提出了那个经典的问题：“小Hi和小Ho，你们能不能够判断一段文字（原串）里面是不是存在那么一些……特殊……的文字（模式串）？”
小Hi和小Ho仔细思考了一下，觉得只能想到很简单的做法，但是又觉得既然河蟹先生这么说了，就肯定不会这么容易的让他们回答了，于是他们只能说道：“抱歉，河蟹先生，我们只能想到时间复杂度为（文本长度 * 特殊文字总长度）的方法，即对于每个模式串分开判断，然后依次枚举起始位置并检查是否能够匹配，但是这不是您想要的方法是吧？”
河蟹点了点头，说道：”看来你们的水平还有待提高，这样吧，如果我说只有一个特殊文字，你能不能做到呢？“
小Ho这时候还有点晕晕乎乎的，但是小Hi很快开口道：”我知道！这就是一个很经典的模式匹配问题！可以使用KMP算法进行求解！“
河蟹满意的点了点头，对小Hi说道：”既然你知道就好办了，你去把小Ho教会，下周我有重要的任务交给你们！“
”保证完成任务！”小Hi点头道。
提示一：KMP的思路
提示二：NEXT数组的使用
提示三：如何求解NEXT数组

输入

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

输出

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

样例输入

5
HA
HAHAHA
WQN
WQN
ADA
ADADADA
BABABB
BABABABABABABABABB
DAD
ADDAADAADDAAADAAD

样例输出

3
1
3
1

0

题目就是使用KMP算法求解匹配数，刚开始的时候TLE，后来想了一下就过了附上，就是在判断完最后一个匹配之后，要重新回溯的时候，不应该使用 i - j + 2，而是假设最后一个不匹配，然后使得较短的串进行回溯，大家一对比红色部分代码就可以很快看出来了

TLE的代码

<span style="color:#333333;">import java.io.File;
import java.io.FileInputStream;
import java.util.Scanner;

public class Main {

public static void main(String[] args) throws Exception {
int n;
String sour, dest;
Scanner cin;
cin = new Scanner(System.in);
//        FileInputStream ins = new FileInputStream(new File("src/in"));cin = new Scanner(ins);
n = cin.nextInt();
while (n-- != 0) {
sour = cin.next();
dest = cin.next();

int next[] = getNext(sour);
int i = 0, j = 0; // i 指定 dest, j 指定sour
int ans = 0;
while(i < dest.length()){
while(i < dest.length() && j < sour.length()){
if(j == -1 || dest.charAt(i) == sour.charAt(j)){
++i;
++j;
}
else{
j = next[j];
}
}
<strong>if(j == sour.length()){
ans ++;
i = i - j + 2 ;
j = 0;
}</strong>
}
System.out.println(ans);
}

}

public static int[] getNext(String str) {
char[] chs = str.toCharArray();
int next[] = new int[chs.length];
next[0] = -1;
int j = 0, k = -1;
int len = chs.length - 1;
while(j < len){
if(k == -1 || chs[j] == chs[k]){
++j; ++k;
next[j] = k;
}
else {
k = next[k];
}
}
return next;
}
}
</span>

Accept代码

<span style="color:#333333;">import java.io.File;
import java.io.FileInputStream;
import java.util.Scanner;

public class Main {

public static void main(String[] args) throws Exception {
int n;
String sour, dest;
Scanner cin;
cin = new Scanner(System.in);
// FileInputStream ins = new FileInputStream(new File("src/in"));cin = new Scanner(ins);
n = cin.nextInt();
while (n-- != 0) {
sour = cin.next();
dest = cin.next();

int next[] = getNext(sour);
int i = 0, j = 0; // i 指定 dest, j 指定sour
int ans = 0;
while (i < dest.length()) {
while (i < dest.length() && j < sour.length()) {
if (j == -1 || dest.charAt(i) == sour.charAt(j)) {
++i;
++j;
} else {
j = next[j];
}
}
<strong>  if (j == sour.length()) {
ans++;
if (i < dest.length()) {
--i;
j = next[j - 1];
}
}</strong>
}
System.out.println(ans);
}

}

public static int[] getNext(String str) {
char[] chs = str.toCharArray();
int next[] = new int[chs.length];
next[0] = -1;
int j = 0, k = -1;
int len = chs.length - 1;
while (j < len) {
if (k == -1 || chs[j] == chs[k]) {
++j;
++k;
next[j] = k;
} else {
k = next[k];
}
}
return next;
}
}
</span>