HDU1686 计算模式串匹配的次数

题目大意：

输入一个T，表示有T组测试数据；

每组测试数据包括一个字符串W，T，T长度大于W小于1000000，w长度小于10000，计算W匹配到T中成功的次数；

 

这题很明显要用KMP算法，不然很容易超时，但在使用kmp算法时也要注意，我第一次将匹配成功的位置得到后，循环进入kmp算法，从前一个匹配到的位置开始算起，但是超时了。后来问完学长之后，清楚了，没必要循环进入kmp，直接一次可以搞定，每次模式串匹配到末尾时，都将计数+1，然后根据next[m]，重新得到j继续与原串进行匹配直到进行到原串结束。

 

代码如下：

#include <iostream>
#include<cstdio>
#include<cstring>
#include<string>
using namespace std;

char b[10100];
char a[1000100];
int next[10100];
int n,m,count;

void getNext(char *b)
{
    int j;
    next[0]=0,next[1]=0;
    for(int i=1;i<m;i++){
        j=next[i];
        while(j&&b[j]!=b[i]) j=next[j];
        if(b[j]==b[i]) next[i+1]=j+1;
        else next[i+1]=0;
    }
}

void kmp(char *a,char *b,int *next)
{
    int j=0;
    for(int i=0;i<n;i++){

        while(j&&a[i]!=b[j]) j=next[j];
        if(a[i]==b[j]) j++;
        if(j==m) count++,j=next[j];
    }
}

int main()
{
    int T;
    cin>>T;
    while(T--){
        count=0;
        string W,T;
        cin>>W>>T;
        n=T.size();
        m=W.size();
        for(int i=0;i<n;i++){
            a[i]=T.at(i);
        }
        for(int i=0;i<m;i++){
            b[i]=W.at(i);
        }

        getNext(b);
        kmp(a,b,next);
        cout<<count<<endl;
    }
    return 0;
}