luogu P2375( KMP 难题 )

 题意：下课前，园长提出了一个问题：“KMP算法只能求出next数组。我现在希望求出一个更强大num数组一一对于字符串S的前i个字符构成的子串，既是它的后缀同时又是它的前缀，并且该后缀与该前缀不重叠，将这种字符串的数量记作num[i]。例如S为aaaaa，则num[4] = 2。这是因为S的前4个字符为aaaa，其中a和aa都满足性质‘既是后缀又是前缀’，同时保证这个后缀与这个前缀不重叠。而aaa虽然满足性质‘既是后缀又是前缀’，但遗憾的是这个后缀与这个前缀重叠了，所以不能计算在内。同理，num[1] = 0,num[2] = num[3] = 1,num[5] = 2。” 计算字符串所有num数组的乘积。

思路：我们处理出g数组，代表从1-i位置前后缀相同的方案总数（不考虑是否重叠）。然后我们模仿KMP求Fail数组的递推思想。加入一个字符时我们将指针向前移动，知道满足num的条件即可。

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const LL mod = 1000000007;
const int maxn = 1000005;
char str[maxn];
int f[maxn];
void get_fail( char* P,int *f ){
    f[0] = -1;
    int m = strlen(P);
    for( int i = 1;i < m;i++ ){
        int j = f[i-1];
        while( j != -1 && P[j+1]!=P[i] )j = f[j];
        if( P[j+1]==P[i] ) f[i] = j+1;
        else f[i] = -1;
    }
}
LL ans;
int g[maxn];
LL mul( LL x,LL y ){
    x *= y;
    return x%mod;
}
void find( char* ss ){
    int p = -1;
    int len = strlen(ss);
    for( int i = 0;i < len;i++ ){
        while(  p != -1 &&( (p+1)*2+2 > i+1 || ss[p+1] != ss[i] ) ) p = f[p];
        if( (p+1)*2+2 <= i+1 &&ss[p+1]==ss[i]  ) p++;
        if( p >= 0 )ans = mul( ans,g[p]+2 );
    }
}
int main(){
    int T;
    scanf("%d",&T);
    while(T--){
        ans = 1;
        scanf("%s",str);
        int len = strlen(str);
        memset( g,0,sizeof(int)*len+1 );
        get_fail( str,f );
        for( int i = 0;i < len;i++ ){
            if(f[i]>=0)
            g[i] = g[ f[i] ] + 1;
        }
        find(str);
        printf("%lld\n",ans);
    }
    return 0;
}