hihocoder 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
ADADADAABABABABABB
DAD
ADDAADAADDAAADAAD

样例输出

3
1
3
1
0

代码实现

#include <iostream>
#include <string>


using namespace std;
void calculateNext(int * nextSZ,const string msString)
{
	int q,k;//q:模版字符串下标；k:最大前后缀长度
	int m = msString.length();//模版字符串长度
    nextSZ[0] = 0;//模版字符串的第一个字符的最大前后缀长度为0
    for (q = 1,k = 0; q < m; ++q)//for循环，从第二个字符开始，依次计算每一个字符对应的next值
    {
        while(k > 0 && msString[q] != msString[k])//递归的求出P[0]···P[q]的最大的相同的前后缀长度k
            k = nextSZ[k-1];          //不理解没关系看下面的分析，这个while循环是整段代码的精髓所在，确实不好理解  
        if (msString[q] == msString[k])//如果相等，那么最大相同前后缀长度加1
        {
            k++;
        }
        nextSZ[q] = k;
    }
	//for (int i = 0; i < m; i++)cout<<nextSZ[i];
	//cout<<endl;
}

int KMP(const string msString,const string yString)
{
	int sum=0,i,j;
	int *nextSZ=NULL,msStringLen=msString.length(),yStringLen=yString.length();
	nextSZ=new int [msString.length()];
	

	calculateNext(nextSZ,msString);

	for (i = 0; i < yStringLen; )
	{
		for (j = 0; j < msStringLen &&msString[j]==yString[i+j]; j++);
		if(j==msStringLen){ sum++;i+=(j-nextSZ[j-1]);}
		else
		{
			i+=(j-nextSZ[j-1]);
		}

	}
	return sum;
}
int main(void)
{

    int N;
	string a,b;
	cin>>N;

	while (N--)
	{
		a.clear();b.clear();
		cin>>a>>b;
		cout<<KMP(a,b)<<endl;
	}
	system("PAUSE");
    return 0;
}