BZOJ 3670 [NOI 2014] 动物园

来源：互联网发布：java获取访问者ip 编辑：程序博客网时间：2024/05/21 07:38

题目链接：BZOJ 3670

好像O(N)的做法很神奇。
next[i]表示i失配时下一个该匹配的位置；cnt[i]表示从i位置根据i=next[i]的方式经过cnt[i]次转移到0位置。
这道题先用KMP求出next数组和cnt数组，然后再进行一次匹配。这次匹配，要保证对于i，若它的next[i]=k，那么k应该满足k<=i/2。

#include<cstdio>#include<cstring>#include<iostream>using namespace std;#define LL long long#define mod (1000000007)const int maxn=(int)1e6+10;char s[maxn];int next[maxn];LL cnt[maxn];LL find(){memset(cnt,0,sizeof(cnt));int l=strlen(s),k=-1,i=0;next[0]=-1;while(i<l){if(k==-1 || s[i]==s[k]){i++; k++;next[i]=k;cnt[i]=cnt[k]+1;}else k=next[k];}LL ans=1; k=-1; i=0;while(i<l){if(k==-1 || s[i]==s[k]){i++; k++;while(k>(i>>1))k=next[k];ans=(ans*(cnt[k]+1))%mod;}else k=next[k];}return ans;}int main(){int N; scanf("%d",&N);while(N--){scanf("%s",s);cout<<find()<<endl;//这题在BZOJ上cout竟然比printf快}return 0;}

0 0