[HDU 3336]Count the String[kmp]

题意:

给出一个字符串,求其所有前缀串为模式串时匹配成功的次数.

思路:

就和AC自动机那题处理模式串重叠的方法一样,每到一处,回溯到根检查一遍.

#include <cstdio>
using namespace std;
const int MAXW = 200005;
const int MOD = 10007;
char W[MAXW];
int n;
int next[MAXW];
void prekmp(char *B, int *next)
{
    int j = -1;
    next[0] = -1;
    for(int i=1; B[i]; i++)
    {
        while(j != -1 && B[j+1] != B[i]) j = next[j];
        if(B[i] == B[j+1]) j++;
        next[i] = j;
    }
    /*for(int i=0;i<n;i++)
        printf("next[%d] %d\n",i,next[i]);*/
}

int kmp(char *B, int *next)
{
    long long ret = 0;
    for(int i=0; B[i]; i++)
    {
        ret++;
        int tmp = next[i];
        while(tmp!=-1)
        {
            ret++;
            tmp = next[tmp];
        }
        ret %= MOD;
    }
    return ret;
}

int main()
{
    int cas;
    scanf("%d",&cas);
    while(cas--)
    {
        scanf("%d",&n);
        scanf("%s",W);
        prekmp(W,next);
        printf("%d\n",kmp(W,next));
    }
}