hihoCoder 1015 : KMP算法

#1015 : 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Ëã·¨
KMPËã·¨µÄÏë·¨ÊÇ£¬²»Òª°Ñ"ËÑË÷λÖÃ"ÒÆ»ØÒѾ­±È½Ï¹ýµÄλÖ㬼ÌÐø°ÑËüÏòºóÒÆ£¬ÕâÑù¾ÍÌá¸ßÁËЧÂÊ¡£ 
Òƶ¯Î»Êý = ÒÑÆ¥ÅäµÄ×Ö·ûÊý - ¶ÔÓ¦µÄ²¿·ÖÆ¥ÅäÖµ
*/
#include<iostream>
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
using namespace std;
#define N  1000001
#define N2 10001
char str
,pat[N2];

void calNext(char str[],int next[])
{
	int L,i,len;
	
	L=0;//×ǰ׺
	next[1]=0;
	
	len=strlen(str);
	
	for(i=2;i<len+1;i++)
	{
		while(L>0&&str[L]!=str[i-1])
			L=next[L];
		if(str[i-1]==str[L])
			L++;
		next[i]=L;
	}
}

int KMP(char str[],char pat[])
{
	int len1,len2,k,sum,i;
	int next[N2];
	len1=strlen(str);
	len2=strlen(pat);
	
	calNext(pat,next);
	k=0;
	sum=0;
	for(i=0;i<len1;i++)
	{
		while(k>0&&str[i]!=pat[k])
			k=next[k];
		if(pat[k]==str[i])
			k++;
		if(k==len2)
		{
			sum++;
			k=next[k];
		}
	}
	return sum;
}

int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%s",pat);
		scanf("%s",str);
		printf("%d\n",KMP(str,pat));
	}
	return 0;
}