BZOJ 1511: [POI2006]OKR-Periods of Words

Description

求一个最长周期.

Sol

KMP.

一个点的最短周期就是 (i-next[i])

此外 (i-next[next[i]],i-next[next[next[i]]]) 等等都是它的周期,知道第一个为0的位置.

然后就没有然后了.

Code

/**************************************************************
    Problem: 1511
    User: BeiYu
    Language: C++
    Result: Accepted
    Time:140 ms
    Memory:6172 kb
****************************************************************/
 
#include<cstdio>
#include<iostream>
using namespace std;
 
const int N = 1000005;
 
int n;long long ans;
char s[N];
int f[N];
 
int main(){
    scanf("%d",&n);
    scanf("%s",s+1);
     
    for(int i=2,j=0;i<=n;i++){
        while(j && s[i]!=s[j+1]) j=f[j];
        if(s[i]==s[j+1]) j++;
        f[i]=j;
    }
     
    for(int i=1;i<=n;i++)
        while(f[f[i]]) f[i]=f[f[i]];
     
    for(int i=1;i<=n;i++) if(f[i]) ans=ans+i-f[i];
    cout<<ans<<endl;
    return 0;
}