bzoj 3670 动物园 KMP

最新推荐文章于 2018-08-14 22:55:10 发布

Loi_a

最新推荐文章于 2018-08-14 22:55:10 发布

阅读量390

点赞数 6

分类专栏： ===字符串相关=== KMP

本文链接：https://blog.csdn.net/Loi_a/article/details/52818688

版权

===字符串相关=== 同时被 2 个专栏收录

17 篇文章 0 订阅

订阅专栏

KMP

3 篇文章 0 订阅

订阅专栏

KMP非常神，我们可以顺着fail数组把一个字符串所有前缀等于后缀的子串跑一边，而且顺序是从大到小。例如ababcccabab，fail11=4，fail4=2，fail2=0；对于这道题来说，我们可以顺着每一位的fail跑一遍（当然会T）求num数组，但是我们只要求出跑了多少次后，fail值等于0就可以了，上例sum[11]=sum[4]+1，sum在初始化的时候求出来。统计答案的时候，num[i]=sum[j]，j是顺着fail数组跑，长度<=(i/2)的第一个值；

#include<iostream>
#include<cstdio>
#include<cstring>
#define LL long long
using namespace std;
char s2[1000005];
LL fail[1000005];
LL  sum[1000005];
LL  num[1000005];
void init(int m)
{
    fail[0]=0;fail[1]=0;
    sum[0]=0;sum[1]=1;
    for(int i=2;i<=m;i++)
    {
        int p=fail[i-1];
        while(p&&s2[i]!=s2[p+1])p=fail[p];
        if(s2[p+1]==s2[i]) p++;
        fail[i]=p;
        sum[i]=sum[p]+1;
    }
}
void get(int m)
{
    int p=0;num[0]=0;
    for(int i=2;i<=m;i++)
    {
        while(p&&s2[i]!=s2[p+1])p=fail[p];
        if(s2[i]==s2[p+1]) p++;
        while(p>i/2) p=fail[p];
        num[i]=sum[p];
    }
}
LL ans=1,k=1e9+7;
int main()
{
    int n;scanf("%d",&n);
    while(n--)
    {
        ans=1;
        scanf("%s",s2+1);
        int L=strlen(s2+1);
        init(L);get(L);
        for(int i=1;i<=L;i++)
        {
            ans*=(num[i]+1);
            ans%=k;
        }
        printf("%lld\n",ans);
    }
}