poj 3336 Count the string

给一个字符串，求该字符串的所有前缀在该字符串中出现的次数

分析：首先，每个前缀都至少出现一次，然后，每次用kmp匹配成功的时候，都至少会匹配一个前缀，以abab举例，ab匹配成功的时候的，ab出现一次，如果是abababab呢，那么abab匹配到最后，还多

匹配了一次ab，那么也要计算上

 

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=1e6+5;
const int mod=10007;
int ans;
char s[maxn];
int f[maxn],len;

void getfail(){
    f[0]=f[1]=0;
    for(int i=1;i<len;i++){
        int j=f[i];
        while(j&&s[i]!=s[j])j=f[j];
        f[i+1]=s[i]==s[j]?j+1:0;
        int k=f[i+1];
        while(k){
            ans++;ans%=mod;
            k=f[k];
        }
    }
}

int main(){
    ios::sync_with_stdio(false);
    int t;cin>>t;
    while(t--){
        cin>>len;
        cin>>s;
        ans=len;
        getfail();
        cout<<ans<<endl;
    }
    return 0;
}