SDU 程序设计思维与实践 Week15 【KMP】

ZJM 与纸条

ZJM 的女朋友是一个书法家，喜欢写一些好看的英文书法。有一天 ZJM 拿到了她写的纸条，纸条上的字暗示了 ZJM 的女朋友 想给 ZJM 送生日礼物。ZJM 想知道自己收到的礼物是不是就是她送的，于是想看看自己收到的礼物在纸条中出现了多少次。

输入

第一行输入一个整数代表数据的组数。
每组数据第一行一个字符串 P 代表 ZJM 想要的礼物, 包含英语字符 {‘A’, ‘B’, ‘C’, …, ‘Z’}, 并且字符串长度满足 1 ≤ |P| ≤ 10,000 (|P| 代表字符串 P 的长度).
接下来一行一个字符串 S 代表 ZJM 女朋友的纸条, 也包含英语字符 {‘A’, ‘B’, ‘C’, …, ‘Z’}, 满足 |P| ≤ |S| ≤ 1,000,000.

输出

输出一行一个整数代表 P 在 S中出现的次数.

输入样例

3
BAPC
BAPC
AZA
AZAZAZA
VERDI
AVERDXIVYERDIAN

输出样例

1
3
0

解题代码

#include<iostream>
#include<string>
using namespace std;
int Next[10010],N;
string s, p;
void get_Next(string&ptr,int len) {
	Next[0] = 0;
	for (int i = 1, j = 0; i < len; i++) {
		while (j && ptr[j] != ptr[i]) j = Next[j - 1];
		if (ptr[j] == ptr[i]) j++;
		Next[i] = j;
	}
}
int KMP(string& str, string& ptr) {
	int len1 = str.length();
	int len2 = ptr.length();
	int cnt = 0;
	int j = 0;//matching ptr[j]
	get_Next(ptr, len2);
	for (int i = 0; i < len1; i++) {
		while (j && ptr[j] != str[i]) j = Next[j - 1];
		if (ptr[j] == str[i]) j++;
		if (j == len2) {
			cnt++;
			j = Next[j - 1];
		}
	}
	return cnt;
}
int main() {
	cin >> N;
	while (N--){
		cin >> p;
		cin >> s;
		cout << KMP(s, p) << endl;
	}
}