kmp-洛谷P2375 动物园

https://www.luogu.org/problem/show?pid=2375
因为有一个限制,我们不能简单的处理;
首先,当i满足限制条件时,对于i+1,我们如果用i的限制条件去更新i+1,那么i+1会有错误;
所以我们要先跑一边无限制的kmp,然后再去求出限制的kmp;
限制有两种;
一是限制next数组(p[])
一是限制sum数组;
这个见代码,都是大神帮我调的5555;
还有我的kmp要预处理i=1的情况;
这个好麻烦;

/*#include<iostream>
#include<cstdio>
#include<cstring>
#define Ll long long
using namespace std;
char s[2000001];
Ll p[1000005],sum[1000005],P[1000005];
Ll n,m,ans,mo=1000000007;
void make(char s[]){
    p[1]=0;sum[1]=1;
    for(int i=2;i<=m;i++){
        int j=p[i-1];
        while(j&&s[j+1]!=s[i])j=p[j];

        p[i]=j;sum[i]=sum[j]+1;
       // cout<<i<<' '<<sum[i]<<endl;
    }
}
void make2(char s[]){
    P[1]=0;
    for(int i=2;i<=m;i++){
        int j=p[i-1];
        while((j*2+2>i)||(j&&s[j+1]!=s[i]))j=p[j];
        if(s[j+1]==s[i])j++;
        P[i]=sum[j];
    }
}
int main()
{
    scanf("%lld",&n);
    while(n--){
        scanf("%s",s+1);m=strlen(s+1);
        make(s);
        make2(s);
        ans=1;
        for(int i=1;i<=m;i++)ans=ans*(P[i]+1)%mo;
        printf("%lld\n",ans);
    }
}*/
#include<iostream>
#include<cstdio>
#include<cstring>
#define Ll long long
using namespace std;
char s[2000001];
Ll p[1000005],sum[1000005],P[1000005];
Ll n,m,ans,mo=1000000007;
void make(char s[]){
    p[1]=0;sum[1]=1;
    for(int i=2;i<=m;i++){
        int j=p[i-1];
        while(j&&s[j+1]!=s[i])j=p[j];
        if(s[j+1]==s[i])j++;
        p[i]=j;sum[i]=sum[j]+1;
    }
}
void make2(char s[]){
    P[1]=0;
    int j=0;
    for(int i=2;i<=m;i++){
        if(j*2+2>i)j=p[j];
        while(j&&s[j+1]!=s[i])j=p[j];
        if(s[j+1]==s[i])j++;
        P[i]=sum[j];
    }
}
/*void make2(char s[]){
    P[1]=0;
    for(int i=2;i<=m;i++){
        int j=P[i-1];
        if(j*2+2>i)j=p[j];
        while(j&&s[j+1]!=s[i])j=p[j];
        if(s[j+1]==s[i])j++;
        P[i]=j;
    }
}*/
int main()
{
    scanf("%lld",&n);
    while(n--){
        scanf("%s",s+1);m=strlen(s+1);
        make(s);
        make2(s);
        ans=1;
        for(int i=1;i<=m;i++)ans=ans*(P[i]+1)%mo;
        /*for(int i=1;i<=m;i++)ans=ans*(sum[P[i]]+1)%mo;*/
        printf("%lld\n",ans);
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值