bzoj1511: [POI2006]OKR-Periods of Words

传送门
直接求一下next，之后把所有的next向前找到最后一个非零地方的next。
然后扫一遍对于每个next非零位置的周期来说就是i-new_next[i]
还是之前的那个性质，n-next[i]是最小循环周期，推一下就变成最长了。

#include<cmath>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
using namespace std;
char s[1001001];
int n,j,ne[1001001];
long long ans;
int main(){
    scanf("%d%s",&n,s);
    ne[0]=j=-1;
    for (int i=0;i<n;){
        for (;j!=-1&&s[i]!=s[j];j=ne[j]);
        ne[++i]=++j;
    }
    for (int i=1;i<=n;i++){
        if (!ne[i]) continue;
        while (ne[ne[i]]) ne[i]=ne[ne[i]];
    }
    for (int i=1;i<=n;i++)
        if (ne[i]) ans+=i-ne[i];
    printf("%lld\n",ans); 
}