hdu 3336 KMP+DP

根据kmp原理，改下kmp的代码。。。。略爽。。。。

#include<cstdio>
#include<stdlib.h>
#include<string.h>
#include<string>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<map>
#include<set>
#include<queue>
#include<vector>
#include<iostream>
using namespace std;
#define mod 10007
int d[200009],next[200009];
void getfail(string s,int n)
{   int j=0;
    next[0]=0,next[1]=0;
    for(int i=1;i<n;i++)
    {   j=next[i];
        while(j&&s[i]!=s[j])
            j=next[j];
        if(s[i]==s[j])
            next[i+1]=j+1;
        else
            next[i+1]=0;
    }
}
int kp(string s,int n)
{
    int j=0,res;
    d[0]=1;
    for(int i=1;i<n;i++)
    {   res=0;
        j=next[i];
        while(1){

            while(j&&s[i]!=s[j])
            {
                j=next[j];
            }
            if(s[i]==s[j]){
                res++;
                if(j!=0&&next[j]==0&&s[i]==s[next[j]])
                   res++;
                j=next[j];
         //
            }
            if(j==0)
                break;

        }
        d[i]=(d[i-1]+res+1)%mod;
    }
    return d[n-1];
}
int main()
{
//    freopen("in.txt","r",stdin);
//    freopen("out.txt","w",stdout);
    int t,n;
    scanf("%d",&t);
    while(t--)
    {   string s;
        scanf("%d",&n);
        cin>>s;
        getfail(s,n);
       // for(int i=0;i<n;i++)
       //     printf("%d ",next[i]);

        printf("%d\n",kp(s,n));
     //   for(int i=0;i<n;i++)
        //    printf("%d ",d[i]);
    }

    return 0;
}