【KMP】BZOJ3670-[Noi2014]动物园

【题目大意】
[依然借用别人的概括]给定一个长为L的字符串(L<=100W)，求一个num数组，num[i]表示长度为i的前缀中字符串S’的数量，其中S‘既是该前缀的前缀也是该前缀的后缀，且|S'|*2<=i
【思路】
KMP中next数组的变形。先算一次next数组和dep数组，其中dep数组表示当前前缀经过j=next[j]可以到达-1，这个值其实就是num数组的雏形。然后再进行一次求解next数组，每次前缀不断进行j=next[j]，直到满足|S'|*2<=i。此时的dep[j]就是当前前缀的num值，由于前一次求解next数组预处理，这个值可以在O(1)时间内求解。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
#define mod 1000000007
const int MAXN=1000000+500;
using namespace std;
int n;
char p[MAXN];
int next[MAXN],dep[MAXN],next2[MAXN];
int len;

void getnext()
{
    int i=0,j=-1;
    next[i]=j;
    dep[i]=0;
    while (i<len)
    {
        if (j==-1 || p[i]==p[j])
        {
            dep[++i]=dep[++j]+1;
            next[i]=j;
        }
        else j=next[j];
    }
}

void getans()
{
    LL ans=1;
    int i=0,j=-1;
    while (i<len)
    {
        if (j==-1 || p[i]==p[j])
        {
            i++,j++;
            while (j!=-1 && (j<<1)>i) 
                j=next[j];
            if (j!=-1 && i!=-1) ans=ans*(LL)(dep[j]+1) % mod;
        }
        else j=next[j];
    }
    printf("%lld\n",ans);
    

}

int main()
{
    scanf("%d",&n);
    for (int i=0;i<n;i++)
    {
        scanf("%s",p);
        len=strlen(p);
        getnext();
        getans();
    }
    return 0;
}