[bzoj3670] [NOI2014] [lg2375] 动物园

nxt数组为KMP的next数组
num[i]储存了i前面可以匹配的串的个数。
先在KMP求nxt中顺便求出num
最后再找到对于i的最大的前后缀不重叠的可匹配的j，ans*=(num[j]+1)%1000000007
ans即为答案

#include<cstdio>
#include<cstring>
using namespace std;
char s[1000100];
int nxt[1000100],num[1000100],n;
long long ans;
void doit(char *s){
    int n=strlen(s);
    nxt[0]=-1;
    nxt[1]=0;
    num[0]=0;
    num[1]=1;ans=1;
    for(int i=1,j=0;i<n;i++){
        while(j>=0&&s[i]!=s[j])j=nxt[j];
        nxt[i+1]=++j;
        num[i+1]=num[j]+1;
    }
    for(int i=1,j=0;i<n;i++){
        while(j>=0&&s[i]!=s[j])j=nxt[j];
        j++;
        while((j<<1)>(i+1))j=nxt[j];
        ans=(ans*((long long)num[j]+1))%1000000007;
    }
    printf("%lld\n",ans);
}
int main(){
    scanf("%d",&n);
    while(n--){
        scanf("%s",s);
        doit(s);
    }
}