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;

}

巴特西

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

最新文章

热门文章