BZOJ 1511 [POI2006]OKR-Periods of Words KMP

题意:
求一个串的所有前缀的最长周期长度之和,特别的,周期为自己的串的最长周期长度视作0.
解析:
直接求一下next,之后把所有的next向前找到最后一个非零地方的Next。
然后扫一遍对于每个next非零位置的周期来说就是i-new_next[i]
还是之前的那个性质,n-next[i]是最小循环周期,推一下就变成最长了。
代码:

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define N 1000100
using namespace std;
typedef long long ll;
int n;
int ne[N];
char s[N];
void getnext()
{
    ne[0]=-1;
    int i=0,j=-1;
    while(i<n)
    {
        if(j==-1||s[i]==s[j])
        {
            i++,j++;
            ne[i]=j;
        }else j=ne[j];
    }
}
int main()
{
    #ifndef ONLINE_JUDGE
        freopen("1511.in","r",stdin);
        freopen("1511.out","w",stdout);
    #endif
    scanf("%d",&n);
    scanf("%s",s);
    getnext();
    ll ans=0;
    for(int i=1;i<=n;i++)
    {
        if(!ne[i])continue;
        while(ne[ne[i]])
            ne[i]=ne[ne[i]];
    }
    for(int i=1;i<=n;i++)
    {
        if(ne[i]!=0)
            ans+=i-ne[i];
    }
    printf("%lld\n",ans);
    #ifndef ONLINE_JUDGE
        fclose(stdin);fclose(stdout);
    #endif
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值