HDU 3336 Count the string

http://acm.hdu.edu.cn/showproblem.php?pid=3336

求字符串的前缀在字符串中一共出现多少次。

这个讲解的很清楚，传送门：http://972169909-qq-com.iteye.com/blog/1114968

我用的思想是递推，把next数组向右平移一位，转移方程d[i]=dp[next[i]]+1;

View Code

#include <iostream>

#include <algorithm>

#include <cstring>

#include <queue>

using namespace std ; 

int n;

char a[200010];

int _next[200010],dp[200010];

void Init_()

{

    int i,k;

    i = 0; k = -1; _next[0] = -1;  

    while(i < n){  

        if(k == -1 || a[i] == a[k]){  

            i++;k++;  

            _next[i] = k;   

        }  

        else  

            k = _next[k];  

    }  

}

int main()

{

    int t;

    scanf("%d",&t);

    while(t--)

    {

        scanf("%d%*c",&n);

        scanf("%s",a);

        Init_();

        int sum=0;

        memset(dp,0,sizeof(dp));

        for(int i=1;i<=n;i++)

        {

            dp[i]=dp[_next[i]]+1;

            sum=(sum+dp[i])%10007;

        }

        printf("%d\n",sum);

    }

    return 0;

}