【BZOJ】3670: [Noi2014]动物园（KMP）

题目

传送门：QWQ

 

---

分析

 

像求next一样求num。

第二次求next时加上不超过一半的条件。

时间复杂度： $ \huge O ( n ) $

---

 

代码

 

// luogu-judger-enable-o2
#include <bits/stdc++.h>
using namespace std;
const int MOD=1000000007;
int n, nxt[1000007], num[1000007];
char s1[1000007];
long long ans;

void kmp()
{
    ans=1;
    nxt[0]=-1; num[1]=1;
    for(int i=1,j;i<=n;i++)
    {
        for(j=nxt[i-1]; j>=0 && s1[i]!=s1[j+1]; j=nxt[j]);
        j++; nxt[i]=j;
        num[i]=num[j]+1;
    }
    for(int i=1,j=-1;i<=n;i++)
    {
        for(; j>=0 && s1[i]!=s1[j+1]; j=nxt[j]);
        j++;  //nxt[i]=j;
        for(;(j<<1)>i;j=nxt[j]);
        ans*=num[j]+1;ans%=MOD;
    }
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%s",s1+1);
        n=strlen(s1+1)+1;
        kmp();
        printf("%lld\n",ans);
    }
    return 0;
}

 

 

 

 【洛谷】P3919 【模板】可持久化数组（主席树）