hihoCoder#1015 KMP算法

题目：http://hihocoder.com/problemset/problem/1015
分析：裸的KMP，正好敲了一遍以前的模板，查出来点错。

代码：

#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int Tmax=1000005;
int T,f[Tmax],slen,alen,ans;
char s[Tmax],a[Tmax];
void getfail()
{
    int i=0,j=-1;  //注意getfail的i也从0开始，因为第一次进去就加到1了
    f[0]=-1;
    while(i<alen)
    {
        if(j==-1||a[i]==a[j])
        {
            i++;
            j++;
            f[i]=j;
        }
        else j=f[j];
    }
    return;
}
void KMP()
{
    int i=0,j=0;
    ans=0;slen=strlen(s);alen=strlen(a);
    getfail();
    while(i<slen)
    {
        if(j==-1||s[i]==a[j])
        {
            i++;
            j++;
        }
        else j=f[j];
        if(j==alen) ans++,j=f[j];  //这里别忘了
    }
    return;
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%s%s",a,s);
        KMP();
        printf("%d\n",ans);
    }
    return 0;
}