首页 > 代码库 > BZOJ 3670 NOI2014 动物园 KMP算法

BZOJ 3670 NOI2014 动物园 KMP算法

题意:

给定一个长为L的字符串(L<=100W),求一个num数组,num[i]表示长度为i的前缀中字符串S’的数量,其中S‘既是该前缀的前缀也是该前缀的后缀,且|S‘|*2<=i

求Π(num[i]+1)%1000000007

这就是KMP算法的一个变形。。。首先求出next数组,顺便求出cnt数组,代表长度为i的前缀经过几次fix=next[fix]会得到0,然后重新匹配一次,这次注意当fix*2>i的时候令fix=next[fix]即可

这题有坑 切忌用num数组代替next

此外勿忘开long long

这题就水过去了。。。。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define M 1000100
using namespace std;
typedef long long ll;
ll ans;
char s[M];
int next[M],cnt[M];
void KMP()
{
    int i,fix=0;
    cnt[1]=1;
    for(i=2;s[i];i++)
    {
        while( fix && s[fix+1]!=s[i] )  fix=next[fix];
        if(s[fix+1]==s[i]) fix++;
        next[i]=fix;
        cnt[i]=cnt[fix]+1;
    }
    fix=0;ans=1;
    for(i=2;s[i];i++)
    {
        while( fix && s[fix+1]!=s[i] )  fix=next[fix];
        if(s[fix+1]==s[i]) fix++;
        while( (fix<<1) > i ) fix=next[fix];
        ans*=cnt[fix]+1;
        ans%=1000000007;
    }
}
int main()
{
    int T;
     
    //freopen("zoo.in","r",stdin);
    //freopen("zoo.out","w",stdout);
     
    for(cin>>T;T;T--)
    {
        scanf("%s",s+1);
        KMP();
        cout<<ans<<endl;
    }
     
}


BZOJ 3670 NOI2014 动物园 KMP算法