bzoj3670: [Noi2014]动物园

真是被毒到了。。。 一開始看錯題了。。。

本來我的做法是記録num直接跳p,結果發現會tle(本機對拍都過了嘛,還是隔壁老爺機)

估计是故意卡了。

正解怎麽做呢,我們類似的再找一次p,但不更新,求值的時候直接往前跳,這樣保證了j几乎是小於i/2的。

j>i/2写成j>i-j+1会挂,原来后面那个要带个等于。。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long LL;
const LL mod=1e9+7;

int p[1100000],num[1100000];
char ss[1100000];
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%s",ss+1);
        int len=strlen(ss+1);LL ans=1LL;
        int j=0;p[1]=0;num[1]=1;
        for(int i=2;i<=len;i++)
        {
            while(j>0&&ss[i]!=ss[j+1])j=p[j];
            if(ss[i]==ss[j+1])j++;
            p[i]=j;
            num[i]=num[j]+1;
        }
        
        j=0;
        for(int i=2;i<=len;i++)
        {
            while(j>0&&ss[i]!=ss[j+1])j=p[j];
            if(ss[i]==ss[j+1])j++;
            
            while(j>0&&j>i/2)j=p[j];
            ans=(ans*(LL(num[j]+1)))%mod;
        }
        printf("%lld\n",ans);
    }
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/AKCqhzdy/p/8875393.html