hdu 1686 Oulipo(KMP)

题意：KMP算法查找模式串在主串中出现的次数。

#include<iostream>
#include<stdio.h>
#include<string.h>
using namespace std;

#define MAXN 10005//字符串长度

char W[MAXN];
char T[1000005];

int _next[MAXN];

void GetNext(char t[]) //求next数组
{
    int j,k,len;
    j=0;//从0开始，首先求_next[1]
    k=-1;//比较指针
    _next[0]=-1;//初始值-1
    len=strlen(t);
    while(j<len)
    {
        if(k==-1||t[j]==t[k]) //指针到头了，或者相等
        {
            ++j;
            ++k;
            _next[j]=k;//此句可由优化替代
            /*优化（求匹配位置时可用）
            if(t[j]!=t[k])_next[j]=k;
            else _next[j]=_next[k];
            //*/
        }
        else k=_next[k];
    }
}

int KMPCount(char s[],char t[]) //统计子串在主串中的出现次数，可重叠
{
    int i,j,lens,lent,cnt;
    i=j=0;
    lens=strlen(s);
    lent=strlen(t);
    cnt=0;

    while(i<lens)
    {
        if(j==-1||s[i]==t[j])
        {
            ++i;
            ++j;
        }
        else j=_next[j];
        if(j==lent)
        {
            ++cnt;//每当成功匹配，则计数
            //j=0;  此时j不用置0，为了计算出模式串在主串中重叠的部分
        }
    }
    return cnt;
}

int main()
{
    int TT;

    scanf("%d",&TT);

    while(TT--)
    {
        scanf("%s",W);
        scanf("%s",T);

        GetNext(W);
        printf("%d\n",KMPCount(T,W));
    }
    return 0;
}