HDU 3336

题目链接：

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

题目描述：

　　给定一个字符串，问该字符串的所有前缀在该字符串出现的次数之和。

　　比如说字符串"abab"，它的前缀分别是"a", "ab", "aba", "abab", 其中"a"在原字符串"abab"中出现了两次，同理"ab"出现了两次，"aba"出现了一次，"abab"出现了一次，所以总的出现次数之和就是六啦。

解题思路：

　　同样运用了next数组原理，同时还需要记忆化搜索一番。dp[i]表示以str[i]为结尾的字符串前缀之和，这里的dp转换思想跟next数组生成的思想是一样的，dp[i] = dp[next[i]] + 1。

　　如果想要更详细的了解可以看一下这篇博客，讲得很详细：http://972169909-qq-com.iteye.com/blog/1114968

代码：

#include <iostream>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <map>
#include <vector>
#include <algorithm>
using namespace std;

#define ll long long
#define INF 0x3f3f3f3f3f
#define MAX 200200
#define MOD 10007

int fail[MAX], dp[MAX];

void getFail(char *str)
{
    fail[0] = -1;
    int k = -1, j = 0;
    int len = strlen(str);
    while(j < len) {
        if(k == -1 || str[j] == str[k]) {
            j++; k++;
            fail[j] = k;
        }
        else
            k = fail[k];
    }
}

int main()
{
    //freopen("debug\in.txt", "r", stdin);
    //freopen("CON", "w", stdout);
    int i, j, k;
    int test, n;
    scanf("%d", &test);
    while(test--) {
        char str[MAX];
        scanf("%d%s", &n, str);
        getFail(str);
        dp[0] = 0;
        int ans = 0;
        for(i = 1; i <= n; ++i) {
            dp[i] = dp[fail[i]] + 1;
            ans += dp[i];
            ans %= MOD;
        }
        printf("%d
", ans);
    }
    return 0;
}