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 ;
}