bzoj3670: [Noi2014]动物园 KMP

kmp位置一直用一个数j来表示比较好。。。

/**************************************************************
    Problem: 3670
    User: xujiahe
    Language: C++
    Result: Accepted
    Time:684 ms
    Memory:10068 kb
****************************************************************/
 
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <iostream>
#include <cmath>
using namespace std;
#define maxn 1001000
#define ll long long
char str[maxn];
int len,next[maxn];
int cal[maxn];
void solve()
{
   // memset(cal,0,sizeof(cal));
   // memset(next,0,sizeof(next));
    len=strlen(str);
    next[1]=0;
    cal[1]=1;
    int j=0;
    for(int i=1;i<len;i++)
    {
        while(j&&str[i]!=str[j])
        {
            j=next[j];
        }
        if(str[i]==str[j])
        {
            j++;
        }
        next[i+1]=j;
        cal[i+1]=cal[j]+1;
    }
/*
    for(int i=1;i<=len;i++)
    {
        printf("%d %d\n",next[i],cal[i]);
    }
*/
    ll ans=1;
    j=0;
    for(int i=1;i<len;i++)
    {
        while(j&&str[j]!=str[i] ) j=next[j];
        if(str[j]==str[i]) j++;
        while( (j<<1) > (i+1) ) j=next[j];
        ans*=cal[j]+1;
        ans%=1000000007;
    }
    printf("%lld\n",ans);
}
int main()
{
    int cas;
    scanf("%d",&cas);
    while(cas--)
    {
        scanf("%s",str);
        solve();
    }
    return 0;
}
/*
3
aaaaa
ab
abcababc
*/