Counting the String 动态规划 + KMP

/*
    题目描述：给出一个长度为n(0 < n < 2e6)的字符串s，问其共有多少子串与s的任一前缀相等。
    
    思路：假设字符串下标从1 ~ n ，dp[i]表示以i结尾的字串中包含前缀的个数，f[i]表示长度为i子串前后最大公共长度。
                令p = f[i] ， 则dp[i] = dp[p] + 1
                这样转移的原因是以i结尾的前缀们一定包含在s[1...p]内，而s[1...p]内一切前缀皆为以i结尾的一个子串；
                除此之外，以i结尾的还有子串s[1...i]，所以要加一；
                f数组的求解是KMP原理的精髓
                
    收获：前缀问题可能与KMP有联系
*/
#pragma warning(disable:4786)
#pragma comment(linker, "/STACK:102400000,102400000")
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<stack>
#include<queue>
#include<map>
#include<set>
#include<vector>
#include<cmath>
#include<string>
#include<sstream>
#include<bitset>
#define LL long long
#define FOR(i,f_start,f_end) for(int i=f_start;i<=f_end;++i)
#define mem(a,x) memset(a,x,sizeof(a))
#define lson l,m,x<<1
#define rson m+1,r,x<<1|1
using namespace std;
const int INF = 0x3f3f3f3f;
const int mod = 10007;
const double PI = acos(-1.0);
const double eps=1e-6;
const int maxn = 2e5 + 5;
char s[maxn];
int f[maxn];
LL dp[maxn];
void getFail()
{
    int m = strlen(s);
    f[0] = 0;   f[1] = 0;
    for(int i = 1 ; i<m ; i++){
        int j = f[i];
        while(j && s[i] != s[j])    j = f[j];
        if(s[i] == s[j])    f[i + 1] = j + 1;
        else        f[i + 1] = 0;
    }
}
int main()
{
    int T , m;
    scanf("%d", &T);
    while(T--){
        scanf("%d %s" , &m , s);
        getFail();
        LL ans = 0;
        for(int i = 1 ; i<= m ; i++){
            int p = f[i];
            dp[i] = 1 + dp[p];
            ans = (ans + dp[i]) % mod;
        }
        printf("%lld\n",ans);
    }
    return 0;
}