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;
}