P3435 [POI2006]OKR-Periods of Words

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
char a[1000003];
int n,f[1000003];
long long ans;
int main(){
    scanf("%d",&n); scanf("%s",a);
    int j;
    for(int i=1;i<n;++i){
        j=f[i];
        while(j&&a[i]!=a[j]) j=f[j];
        f[i+1]= a[i]==a[j] ? j+1:0;
    }
    for(int i=1;i<=n;++i){
        if(!f[i]) continue;
        j=f[i];
        while(f[j]) f[i]=j=f[j];
        ans+=i-j;
    }
    /*for(int i=1;i<=n;++i){
        if(!f[i]) continue; 
        int len=i-f[i];
        ans+=i-(i%len ? i%len:len);
    } //并不能用在这题上
    // 8 uuauuauu 反例
    // 24*/
    printf("%lld",ans);
    return 0;
}