【Ybtoj 第8章例题3】周期长度和【KMP】

---

解题思路

手推一下发现如果前缀存在一个长为j的公共前后缀，那么它有一个长为i-j的周期。
$j$可以通过$KMP$得到，周期最长也就是$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);
}