HDU 3336 输出包括从1到len长 字符串前缀的总个数（+DP）

Sample Input
1
4
abab

Sample Output
6
输出包括从1到len长 字符串前缀的总个数
abab：包括2个a，2个ab，1个aba，1个abab

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

char S[200010] ;
int next[200010] ;
int dp[200010] ;
int len ;

void getNext()
{
int j, k;
j = 0; k = -1; next[0] = -1;
while(j < len)
if(k == -1 || S[j] == S[k])
next[++j] = ++k;
else
k = next[k];
}

int main ()
{
int T ;
scanf("%d" , &T);
while (T--)
{
scanf("%d" , &len) ;
scanf("%s" , S) ;
getNext() ;
dp[0] = 0 ;
int ans = 0 ;
int i ;
for (i = 1 ; i<= len ; i++)
{
dp[i] = dp[next[i]] + 1 ;
dp[i] %= 10007 ;
ans += dp[i] ;
ans %= 10007 ;
}
printf("%d\n" , ans) ;

}

return 0 ;
}