学习KMP可以看一下这篇博客(【经典算法】——KMP,深入讲解next数组的求解),写的很好。
代码方面我参考的是kuangbin的模板(ACM博客_kuangbin)。
本题就是套KMP模板匹配一个串T在另一个串S中出现的次数。
代码如下:
#include <cstring>
#include <cstdlib>
#include <cstdio>
using namespace std;
const int N = 1000005;
char S[N], T[N];
int lens, lent;
int Next[N];
void getNext()
{
int i = 0;
int k = -1;
Next[0] = -1;
while (i < lent)
if (k == -1 || T[i] == T[k])
Next[++i] = ++k;
else
k = Next[k];
}
int KMP_count()
{
int i = 0;
int j = 0;
getNext();
int cnt = 0;
while (i < lens)
{
if (j == -1 || S[i] == T[j])
{
i++;
j++;
}
else
j = Next[j];
if (j == lent)
{
cnt++;
j = Next[j];
}
}
return cnt;
}
int main()
{
//freopen("test.txt", "r", stdin);
//freopen("out.txt", "w", stdout);
int t;
scanf("%d", &t);
while(t--)
{
scanf("%s%s", T, S);
lens = strlen(S);
lent = strlen(T);
printf("%d\n", KMP_count());
}
return 0;
}