#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;
}