HDU 3336 Count the string KMP

题目地址：http://acm.hdu.edu.cn/showproblem.php?pid=3336

如果你是ACMer，那么请点击看下

题意：求每一个的前缀在母串中出现次数的总和。

AC代码：

 

#include <iostream>

#include <cstdio>

#include <cstdlib>

#include <cmath>

#include <cstring>

#include <string>

#include <vector>

#include <list>

#include <deque>

#include <queue>

#include <iterator>

#include <stack>

#include <map>

#include <set>

#include <algorithm>

#include <cctype>

using namespace std;



typedef long long LL;

const int N=200005;

const LL II=100000000;

const int INF=0x3f3f3f3f;

const double PI=acos(-1.0);



int next[N],c[N],ans,len;

char str[N];



void getNext(char *p)

{

    memset(c,0,sizeof(c));

    int j=0,k=-1;

    next[0]=-1;ans=0;

    while(j<len)//len是p的长度

    {

        if(k==-1||p[j]==p[k])

        {

            if(k!=-1)

            {

                c[j]=c[k]+1;

                ans+=c[j];

            }

            j++;

            k++;

            next[j]=k;

        }

        else

            k=next[k];

    }

}



int main()

{

    int i,j,T;

    cin>>T;

    while(T--)

    {

        scanf("%d%s",&len,str);

        getNext(str);

        printf("%d\n",(ans+len)%10007);

    }

    return 0;

}