HDU 1686 Kmp(字符串匹配)

/*
Kmp算法的运用
这里是要统计字符在母串中出现的次数
只要在原来的基础上稍作修改即可
*/
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
char a[10005], b[1000005];
int next[10005];

void Getnext(char* t)//构造next数组，这是关键
{
int lent = strlen(t), i = 0, j = -1;
memset(next,0,sizeof(int)*lent);
next[0] = -1;
while(i < lent)
{
if(j==-1 || t[i]==t[j])
{
i++;
j++;
next[i] = j;
}
else j = next[j];
}
}

void Kmp(char* a, char* b)//Kmp函数
{
int ans = 0, i = 0, j = 0, lena = strlen(a), lenb = strlen(b);
Getnext(a);
while(i<lenb && j<lena)
{
if(j==-1 || a[j]==b[i])
{
i++;
j++;
}
else j = next[j];
if(j >= lena)
{
ans++;//统计字串出现的次数
j = next[j];
}
}
printf("%d\n",ans);
}

int main()
{
int n;
scanf("%d",&n);
while(n--)
{
scanf("%s",a);
scanf("%s",b);
Kmp(a,b);
}
return 0;
}