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;
}