解题思路
手推一下发现如果前缀存在一个长为j的公共前后缀,那么它有一个长为i-j的周期。
j j j可以通过 K M P KMP KMP得到,周期最长也就是 j j j最小。
代码
#include<iostream>
#include<cstdio>
#include<iomanip>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
char s[1000010];
long long ans,n,i,j,p[1000010];
int find(int x){
//查询最小的j
if(p[x])
return p[x]=find(p[x]);
else return x;
}
int main(){
scanf("%lld",&n);
scanf("%s",s+1);
j=0;
for(i=1;i<n;i++)
{
while(j>0&&s[i+1]!=s[j+1])
j=p[j];
if(s[i+1]==s[j+1])
j++;
p[i+1]=j;
}
for(int i=1;i<=n;i++)//统计答案
ans+=i-find(i);
printf("%lld",ans);
}