BZOJ[3670][Noi2014]动物园 KMP

传送门ber~

定义位置$i$要花$cnt_i$步跳到头，那么$num_i$就等于这$cnt_i$个编号$\le i/2$有多少个

如果对于每一个编号都暴力往回跳的话，显然是很容易被卡掉的

那么再做一遍kmp限制一下长度就好了！

代码如下：

#include<cstring>
#include<cstdio>
#define MOD 1000000007
#define N 1000020
using namespace std;
char s[N];
int nex[N],num[N],T;
inline void Solve(char *s){
    int len=strlen(s+1);
    nex[1]=0;num[1]=1;
    int i=2,j=0;
    long long ans=1;
    while(i<=len){
        if(!j || s[i]==s[j+1]){
            if(s[i]==s[j+1]) j++;
            nex[i]=j;
            num[i]=num[j]+1;
            i++;
        }
        else j=nex[j];
    }
    i=2,j=0;
    while(i<=len){
        if(!j || s[i]==s[j+1]){
            if(s[i]==s[j+1]) j++;
            while(j*2>i) j=nex[j];
            ans=ans*(num[j]+1);
            ans=ans%MOD;
            i++;
        }
        else j=nex[j];
    }
    printf("%lld\n",ans);
}
int main(){
    scanf("%d",&T);
    while(T--){
        scanf("%s",s+1);
        Solve(s);
    }
return 0;
}