HDU 3336 KMP_NEXT

题意：一个字符串，长度是n，那么他一共有n个前缀，现在问，这个前缀在原串中有多少个，将所有的和加起来取模。

例如：abab

那么他的前缀有a , ab ,aba , abab。

对应的在原串中的个数为a有2个，ab有2个，aba有1个，abab有1个。

那么总数是6。

思路：对这个字符串求出Next数组，那么对于每个Next[i] = k ,那么我们就知道前缀0 -> k - 1处有一个匹配。那么累加即可。

那么只需根据Next数组匹配成功的个数进行累加即可。

#include <set>
#include <map>
#include <stack>
#include <cmath>
#include <queue>
#include <cstdio>
#include <string>
#include <vector>
#include <iomanip>
#include <cstring>
#include <iostream>
#include <algorithm>
#define Max 2505
#define FI first
#define SE second
#define ll long long
#define PI acos(-1.0)
#define inf 0x3fffffff
#define LL(x) ( x << 1 )
#define bug puts("here")
#define PII pair<int,int>
#define RR(x) ( x << 1 | 1 )
#define mp(a,b) make_pair(a,b)
#define mem(a,b) memset(a,b,sizeof(a))
#define REP(i,s,t) for( int i = ( s ) ; i <= ( t ) ; ++ i )

using namespace std;

#define N 222222
int Next[N] ;
char a[N] ;
void get_Next(){
    Next[0] = -1 ;
    int k = -1 , j = 0 ;
    int l = strlen(a) ;
    while(j < l && k < l){
        if(k == -1 || a[k] == a[j]){
            k ++ ; j ++ ;
            Next[j] = k ;
        }
        else k = Next[k] ;
    }
}
int n ;
int num[N] ;
#define MOD 10007
void solve(){
    cin >> n ;
    cin >> a ;
    mem(num ,0) ;
    get_Next() ;
    int l = strlen(a) ;
    for (int i = 0 ; i <= l ; i ++ ){
        if(Next[i] == -1 || Next[i] == 0)continue ;
        num[Next[i] - 1] ++ ;
    }
    int ans = l ;
    for (int i = 0 ; i < l ; i ++ ){
        ans = (ans + num[i]) % MOD ;
    }
    printf("%d\n",ans) ;
}
int main() {
    int _ ; cin >> _ ; while( _ -- )solve() ;
    return 0 ;
}