HDU 3336 Count the String（KMP+DP）

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

题意：给出一个字符串，计算所有前缀在字符串中出现的次数。

思路：考虑KMP的next[]来解题。next[i]=j表示最大的j使得0~j==i-j~i。

        对于样例的next[]分析如下：

                  0   1   2   3

      a[]        a    b   a   b

      next[]   -1   0   0   1

     dp[i]表示子串a[0~i]共含有以a[i]为结尾的前缀的数目，则以a[i]结尾的前缀数就是自己本身加上以a[next[i]]结尾的前缀数,即dp[next[i]]。

     

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

const int maxn = 200000 + 5;

int n;

char a[maxn];
int next[maxn];
int dp[maxn];

void get_next()
{
    int i = -1, j = 0;
    ::next[0] = -1;
    int l = strlen(a);
    while (j < l)
    {
        if (i == -1 || a[i] == a[j])
            ::next[++j] = ++i;
        else
            i = ::next[i];
    }
}

int main()
{
    //freopen("D:\\txt.txt", "r", stdin);
    int t;
    cin >> t;
    while (t--)
    {
        cin >> n;
        cin >> a;
        get_next();
        memset(dp, 1, sizeof(dp));
        dp[0] = 0;
        int sum = 0;
        for (int i = 1; i <= n; i++)
        {
            dp[i] = dp[::next[i]] + 1;
            sum += dp[i] % 10007;
        }
        cout << sum % 10007 << endl;
    }
    return 0;
}