POJ 3461 KMP

题意：输入两个字符串，在第二个字符串中含有多少个子串与第一个字符串相同。

改写写KMP算法

#include <iostream>
#include<cstring>
#include<cstdio>
using namespace std;
int ans;

char sa[10010],sb[1000001];
int fail[10010];
int kmp(char* str, char* pat)
{
int i, j, k;
memset(fail, -1, sizeof(fail));
for (i = 1; pat[i]; ++i)
{
for (k=fail[i-1]; k>=0 && pat[i]!=pat[k+1];
k=fail[k]);
if (pat[k + 1] == pat[i]) fail[i] = k + 1;
}
i = j = 0;
while( str[i] ) // By Fandywang
{
if( pat[j] == str[i] ) ++i, ++j;
else if(j == 0)++i;//第一个字符匹配失败，从str下个字符开始
else j = fail[j-1]+1;

if(!pat[j])
{
ans++;
j=fail[j-1]+1;
}
}
}
int main()
{
int n;
scanf("%d",&n);
while(n--)
{
scanf("%s%s",sa,sb);
ans=0;
kmp(sb,sa);

cout<<ans<<endl;
}
return 0;
}