小Hi和小Ho是一对好朋友,出生在信息化社会的他们对编程产生了莫大的兴趣,他们约定好互相帮助,在编程的学习道路上一同前进。
这一天,他们遇到了一只河蟹,于是河蟹就向小Hi和小Ho提出了那个经典的问题:“小Hi和小Ho,你们能不能够判断一段文字(原串)里面是不是存在那么一些……特殊……的文字(模式串)?”
小Hi和小Ho仔细思考了一下,觉得只能想到很简单的做法,但是又觉得既然河蟹先生这么说了,就肯定不会这么容易的让他们回答了,于是他们只能说道:“抱歉,河蟹先生,我们只能想到时间复杂度为(文本长度 * 特殊文字总长度)的方法,即对于每个模式串分开判断,然后依次枚举起始位置并检查是否能够匹配,但是这不是您想要的方法是吧?”
河蟹点了点头,说道:”看来你们的水平还有待提高,这样吧,如果我说只有一个特殊文字,你能不能做到呢?“
小Ho这时候还有点晕晕乎乎的,但是小Hi很快开口道:”我知道!这就是一个很经典的模式匹配问题!可以使用KMP算法进行求解!“
河蟹满意的点了点头,对小Hi说道:”既然你知道就好办了,你去把小Ho教会,下周我有重要的任务交给你们!“
”保证完成任务!”小Hi点头道。
Input
第一行一个整数N,表示测试数据组数。
接下来的N*2行,每两行表示一个测试数据。在每一个测试数据中,第一行为模式串,由不超过10^4个大写字母组成,第二行为原串,由不超过10^6个大写字母组成。
其中N<=20
Output
对于每一个测试数据,按照它们在输入中出现的顺序输出一行Ans,表示模式串在原串中出现的次数。
Sample Input
5 HA HAHAHA WQN WQN ADA ADADADA BABABB BABABABABABABABABB DAD ADDAADAADDAAADAADSample Output
3 1 3 1 0
题目分析:KMP模板题,但我在做的时候一直TLE,有一个点需要注意下,在代码中标注了。
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string>
#include <cstring>
#define MAXN 10004
using namespace std;
char p[MAXN],s[MAXN*100];
int nex[MAXN];
void get_nex(){
int k = -1;
nex[0] = -1;
int len = strlen(p);
for(int i=1;i<len;++i){
while(k > -1 && p[k+1] != p[i]){
k = nex[k];
}
if(p[k+1] == p[i]) ++k;
nex[i] = k;
}
}
int solve(){
get_nex();
int ans = 0;
int k = -1;
int len = strlen(s);
for(int i=0;i<len;++i){
while(k > -1 && p[k+1] != s[i]){
k = nex[k];
}
if(p[k+1] == s[i]) ++k;
if(k == strlen(p)-1){
++ans;
k = nex[k];
//k = -1; //这个地方需要注意下,我之前一直卡在这个点,被注释掉的是之前的一个做法,像这
//i = i - strlen(p) + 1; //个题就卡你这个点,其实可以优化为k=nex[k]一句,具体解释我加到了转载的一篇讲KMP的文章里
}
}
return ans;
}
int main(){
int n;
scanf("%d",&n);
for(int i=0;i<n;++i){
scanf("%s%s",p,s);
cout << solve() << endl;
}
return 0;
}