bzoj3670: [Noi2014]动物园
真是被毒到了。。。 一開始看錯題了。。。
本來我的做法是記録num直接跳p,結果發現會tle(本機對拍都過了嘛,還是隔壁老爺機)
估计是故意卡了。拍了好久QWQ
正解怎麽做呢,我們類似的再找一次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; }
pain and happy in the cruel world.