#5. 【NOI2014】动物园

http://uoj.ac/problem/5

题目即题解

肯定是 KMP 了(连着两年字符串 KMP SA  今年AC自动机?SAM?)

然后求长度不超过一半的next

直接用nnext暴力 会T的很惨 比如 aaaaaaaaaaaaa

然后可以记录两个 next 一个正常 一个是半大的 每次用半大的更新(结论自己想想)  如果长度太大 用正常的转移减少长度

剩下的数目自己维护下就好了

#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define LL long long

const int INF = 1e9;
const int MO = 1000000007;
const int MAXN = 1e6+10;

char s[MAXN];
int n;
int fail[MAXN];
int nnext[MAXN];
int num[MAXN];

int solve()
{
	int ans=1;
	int len=strlen(s+1);
	num[0]=0;
	num[1]=1;
	for(int i=2;i<=len;i++)
	{
		int now=fail[i-1];
		while(now!=0&&s[now+1]!=s[i]) now=fail[now];
		if(s[now+1]==s[i]) fail[i]=now+1;
		else fail[i]=0;
		
		now=nnext[i-1];
		while(now!=0&&s[now+1]!=s[i]) now=nnext[now];
		if(s[now+1]==s[i]) nnext[i]=now+1;
		else nnext[i]=0;
		
		while (nnext[i]*2>i) nnext[i]=fail[nnext[i]];
		
		num[i]=num[fail[i]]+1;
	//	cout<<fail[i]<<' ';
		ans=(LL)ans*(num[nnext[i]]+1)%MO;
	}//cout<<endl;
	return ans;
}

int main()
{
	scanf("%d",&n);
	while(n--)
	{
		memset(fail,0,sizeof(fail));
		memset(nnext,0,sizeof(nnext));
		memset(s,0,sizeof(s));
		memset(num,0,sizeof(num));

		scanf("%s",s+1);
		printf("%d\n",solve());
	}
	return 0;
}