HIHO code 1015 KMP算法 【KMP OR EXMP】

#1015 : KMP算法

Time Limit: 1000ms

Case Time Limit: 1000ms

Memory Limit: 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，表示模式串在原串中出现的次数。

Sample Input

5
HA
HAHAHA
WQN
WQN
ADA
ADADADA
BABABB
BABABABABABABABABB
DAD
ADDAADAADDAAADAAD

Sample Output

3
1
3
1
0

Emacs Normal Vim

KMP不好想，用好想的EXKMP水一遍；

AC代码：

#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;

const int MAXN=1e6+33;

char S[MAXN],T[MAXN];
int ne[MAXN],ex[MAXN];

void Get_ne(char *T,int *ne,int LT)
{
	ne[0]=LT; 
	int i=0;
	while(i+1<LT&&T[i]==T[i+1]) ++i;
	ne[1]=i; int id=1;
	for(i=2;i<LT;++i)
	{
		int mx=id+ne[id]-1,ll=ne[i-id];
		if(i+ll-1>=mx) {
			 int j=max(0,mx-i+1);
			 while(i+j<LT&&T[j]==T[i+j]) ++j;
			 ne[i]=j; id=i; 
		}
		else ne[i]=ll;
	}
}
int EXKMP(char *S,char *T,int LS,int LT)
{
	int i=0;
	while(i<LS&&i<LT&&S[i]==T[i]) ++i;
	ex[0]=i;  int id=0;
	for(i=1;i<LS;++i)
	{
		int mx=id+ex[id]-1,ll=ne[i-id];
		if(i+ll-1>=mx) {
			int j=max(0,mx-i+1);
			while(j<LT&&i+j<LS&&T[j]==S[i+j]) ++j;
			ex[i]=j; id=i;
		} 
		else ex[i]=ll;
	}
	int ans=0;
	for(i=0;i<LS;++i) 
	{
		if(ex[i]==LT) ans++; 
	}
	return ans;
}
int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%s %s",T,S);
		int LT=strlen(T),LS=strlen(S);
		Get_ne(T,ne,LT);
		int ans=EXKMP(S,T,LS,LT);
		printf("%d\n",ans);
	}
	return 0;
 }