【NOI2014】动物园

题目大意：

对于字符串 S 的前 i 个字符构成的子串，既是它的后缀同时又是它的前缀，并且该后缀与该前缀不重叠，将这种字符串的数量记作num[i]。例如 S 为 aaaaa，则 num[4] = 2。这是因为 S 的前 4 个字符为 aaaa，其中a 和 aa 都满足性质‘既是后缀又是前缀’，同时保证这个后缀与这个前缀不重叠。特别地，为了避免大量的输出，你不需要输出 num[i] 分别是多少，你只需要输出∏ (num[i] + 1) 对 1,000,000,007 取模的结果即可。其中∏ (num[i] + 1)=1= (num[1] + 1) × (num[2] + 1) × ⋯ × (num[L] + 1)。

思路：

这题可以想到kmp，然后同样的做法在里面统计一下数量就好了。

程序：

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define ll long long
#define N 1000005
#define MOD 1000000007
using namespace std;

int len,next[N],d[N];
char s[N];

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

int main()
{
    int t;
    scanf("%d",&t);
    while (t--)
    {
        scanf("%s",s);
        len=strlen(s);
        for (int i=len;i>=1;i--)
            s[i]=s[i-1];
        getnext();
        int ans=1,i=1,j=0;
        while (i<=len)
            if (s[i]==s[j+1]||j==-1)
            {
                j++;
                if (j>0&&j>i/2) j=next[j];
                ans=(ll)ans*(d[j]+2)%MOD;
                i++;
            }
            else j=next[j];
        printf("%d",ans);
        if (t) cout<<endl;
    }
    return 0;
}