bzoj 1511: [POI2006]OKR-Periods of Words【kmp】

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\n",ans);
    return 0;
} 
posted @ 2018-09-08 20:20  lokiii  阅读(129)  评论(0编辑  收藏  举报