P2375 [NOI2014]动物园

//Pro: P2375 [NOI2014]动物园

//题意：求1<=j<=i/2, s[1..j]==s[i/2+j]的j的个数

//考虑一下可以重复的前缀后缀的个数，就是nxt一直往前跳的次数
//那么不重复的个数就是当nxt跳到<=i/2的时候，nxt还能继续往前跳的次数
//所以记录i能跳nxt直到nxt=0的次数，记为num[i]
//那么在统计i的答案的时候，就让j一直往nxt跳，直到j<=i/2，然后让ans*=(num[j]+1) 

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

const int N=1e6+5;
const int mod=1e9+7;

int n,len;
int nxt[N],num[N];
char s[N];

long long ans;
int main()
{
    scanf("%d",&n);
    while(n--)
    {
        scanf("%s",s+1);
        len=strlen(s+1);
//        memset(num,0,sizeof(num));
//        memset(nxt,0,sizeof(nxt));
        ans=1;
        num[1]=1;
        for(int i=2,j=0;i<=len;++i)
        {
            while(j&&s[j+1]!=s[i])
                j=nxt[j];
            if(s[j+1]==s[i])
                ++j;
            nxt[i]=j;
            num[i]=num[j]+1;
        }
//        for(int i=1;i<=len;++i)
//            printf("%d ",num[i]);
//        puts("");
        for(int i=2,j=0;i<=len;++i)
        {
            while(j&&s[j+1]!=s[i])
                j=nxt[j];
            if(s[j+1]==s[i])
                ++j;
            while((j<<1)>i)
                j=nxt[j];
            (ans*=1ll*(num[j]+1))%=1ll*mod;
        }
        printf("%lld
",ans);
    }
    return 0;
}