BZOJ 3670 [NOI 2014] 动物园

题目链接：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;
}