Week15—KMP例题，模式串在子串中出现的次数

问题描述

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

问题分析

暴力匹配：枚举 i = 0, 1, 2 … , n-m，将 S[i…i+m] 与 P 作比较。如果一致，则找到了一个匹配，这种方法的复杂度为O(nm),超时！
KMP：
做法： 使用KMP算法，暴力算法O(nm)复杂度易超时，KMP算法，在根据 Next 数组转移的过程中，如果模式串匹配到的长
度等于模式串的长度，则成功匹配，匹配成功以后根据 Next 数组进行转移，开始下一轮匹配

代码

#include<iostream>
#include<cstring>
using namespace std;
int Next[10010];
void getNext(const char ptr[],int len)
{
	Next[0]=0;
	for(int i=1,j=0;i<len;++i)
	{
		while(j&&ptr[i]!=ptr[j])
			j=Next[j-1];
		if(ptr[i]==ptr[j]) j++;
		Next[i]=j;	
	}
}

int KMP(const char str[],const char ptr[])
{
	int len1=strlen(str);
	int len2=strlen(ptr);
	int cnt =0;//cnt记录成功匹配的次数
	getNext(ptr,len2);
	for(int i=0,j=0;i<len1;++i)
	{
		while(j&&str[i]!=ptr[j])
			j=Next[j-1];
		if(str[i]==ptr[j]) j++;
		if(j==len2)
		{//匹配到模式串末尾 
			cnt++;
			j=Next[j-1];//根据Next数组转移 
		}
	} 
	return cnt; 
}
int main()
{
	int n;
	cin>>n;//组数

	for(int i=1;i<=n;i++)
	{	char str[1000010],ptr[10010];
		memset(Next,0,sizeof(Next));
		cin>>ptr;
		getchar();
		cin>>str;
		int ans=KMP(str,ptr);
		cout<<ans<<endl;
	} 
}