bzoj 1511: [POI2006]OKR-Periods of Wordskmp

Posted 2021-01-05 lokiii

n-ne[n]是n的最长循环节长度，其实就是n-最短前缀=后缀长度
然后我们要求最短循环节，其实就是ne一直往前跳，跳到不能跳为止，这时的n-ne[n]就是n的最短循环节长度

#include<iostream>
#include<cstdio>
using namespace std;
const int N=1000005;
int n,ne[N];
long long ans;
char s[N];
void gtne()
{
    int i=0,j=-1;
    ne[0]=-1;
    while(i<n)
    {
        if(s[i]==s[j]||j==-1)
            ne[++i]=++j;
        else
            j=ne[j];
    }
}
int main()
{
    scanf("%d%s",&n,s);
    gtne();
    for(int i=1;i<=n;i++)
        if(ne[i])
            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
",ans);
    return 0;
}