学习成神之路

爱情和编程,你选择哪个

【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;
}
阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq872425710/article/details/79249215
个人分类: kmp 纪中成神之路
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭