BZOJ3670: [Noi2014]动物园

t<=5次询问每次问一个<=1e6的串的$\sum_{i=1}^{n} (num_i+1)$，其中num[i]表示既是前缀i的前缀又是前缀i的后缀且这两部分不重叠的子串的数量。

方法一：在KMP树上倍增！吃枣药丸+tle+制杖

方法二：开另一个数组记下每个点在KMP树中的深度，另外开一个指针记下“前后不重叠”的匹配不就完了。。

#include<string.h>
#include<stdlib.h>
#include<stdio.h>
//#include<assert.h>
#include<algorithm>
//#include<iostream>
using namespace std;

int t,n;
#define maxn 1000011
char s[maxn];int fail[maxn],cnt[maxn];
const int mod=1e9+7;
int main()
{
    scanf("%d",&t);
while (t--)
{
    scanf("%s",s+1); n=strlen(s+1);
    int f1=0,f2=0; fail[1]=0; cnt[0]=-1; cnt[1]=0;
    int ans=1;
    for (int i=2;i<=n;i++)
    {
        while (f1 && s[i]!=s[f1+1]) f1=fail[f1];
        if (s[i]==s[f1+1]) f1++;
        fail[i]=f1; cnt[i]=cnt[f1]+1;
        
        while (f2 && s[i]!=s[f2+1]) f2=fail[f2];
        if (s[i]==s[f2+1]) f2++;
        while (f2 && f2*2>i) f2=fail[f2];
        ans=1ll*ans*(cnt[f2]+2)%mod;
    }
    printf("%d\n",ans);
}
    return 0;
}