hdu3336_VJkuangbin16kmp_Count the string

题目大意：给定字符串长度m和字符串p，求出该字符串的前缀总数+每个前缀在p中出现的次数

(1)思考：
①(位置编号1~m)一个以i结尾的前缀，若能在p中找到匹配，且设该匹配的结尾是j，那么就等价于p的子串p[1~j]存在这样一个前缀后缀匹配。
②而且p的一个前缀也等价于以该前缀的结尾为子串的一个前缀后缀匹配。(这种情况是i==j)。
③并且可以证明，这样的关系是一一对应的。

(2)问题转化：
①题目所求<=>p中以i结尾的子串的所有前缀后缀匹配个数之和(1<=i<=m)
②dp[i]:以i结尾的子串的前缀后缀匹配个数
则ans = ∑dp(1<=i<=m)

(3)解：
①next[j] = i;
<1>官方解释:p的子串1~j的最大前缀后缀匹配长度为i(i!=j)。
<2>理解:p的两个子串A, B，A的起点是1，B的终点是j，A向右扩展，B向左扩展，要使A == B，能扩展的最大位数是i - 1，即扩展到i(i < j)。
②dp[i]的解法
<1>状态转移方程:
dp[i] = dp[next[i]] + 1, 初始化:dp[i] = 1(1<=i<=m)，即该子串1~i的全长。
<2>解释:
dp[j]的理解:终点是j，起点是i(1<=i<=j)的子串中为p的前缀的数量，所以枚举起点即可。由以上对next的理解知，子串j-next[j]+1~j是p的前缀及与子串1~next[j]匹配，而且起点为2~j-next[j]的子串都无p的前缀，所以只用求子串j-next[j]+1~j的前缀后缀的匹配数。又因为为1~next[j]与子串j-next[j]+1~j相等，所以dp[j] = dp[next[j]] + 1, (1为起点为1的特殊情况)

#include <cstdio>

using namespace std;
const int maxn = 200010;
const int mod = 10007;
int next[maxn], dp[maxn];
char p[maxn];
int m;

void getnext()
{
    int i = 0, j = -1;
    next[0] = -1;
    while(i < m)
    {
        if(j==-1 || p[i]==p[j]) next[++i] = ++j;
        else j = next[j];
    }
}

int main()
{
    int t;
    scanf("%d", &t);
    while(t--)
    {
        scanf("%d", &m);
        scanf("%s", p);
        getnext();
        for(int i = 1; i <= m; ++i) dp[i] = 1;
        dp[0] = 0;
        int sum = 0;
        for(int i = 1; i <= m; ++i)
        {
            dp[i] = dp[next[i]] + 1;
            sum += dp[i] % mod;
            sum %= mod;
        }
        printf("%d\n", sum);
    }
    return 0;
}