BZOJ 2160: 拉拉队排练（马拉车）

[原题链接(https://www.lydsy.com/JudgeOnline/problem.php?id=2160)

分析

马拉车板子。

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int maxn=2e6+105,mod=19930726;
int n;
LL k;
int p[maxn],sum[maxn];
char s[maxn];
LL ans=1;
int qkpow(int a,int p)
{
    LL t=1,tt=a%mod;
    while(p)
    {
        if(p&1)t=t*tt%mod;
        tt=tt*tt%mod;
        p>>=1;
    }
    return t;
}
void manacher(char *s,int n)
{
    int mx=0,id=0;
    for(int i=1;i<=n;i++)
    {
        p[i]=(mx>i)?min(p[2*id-i],mx-i):1;
        while(s[i-p[i]]==s[i+p[i]])p[i]++;
        if(i+p[i]>mx)
        {
            mx=p[i]+i;
            id=i;
        }
    }
}
int main()
{
    //freopen("in.txt","r",stdin);
    scanf("%d%lld",&n,&k);
    scanf("%s",s+1);
    s[0]='$';
    manacher(s,n);
    for(int i=1;i<=n;i++)
        sum[2*p[i]-1]++;
    for(int i=n-1;i;i--)
        sum[i]+=sum[i+2];
    for(int i=n;i>=1;i--)
    {
        if(sum[i]<k)
        {
            k-=sum[i];
            ans=ans*qkpow(i,sum[i])%mod;
        }
        else
        {
            ans=ans*qkpow(i,k)%mod;
            k=0;
            //break;
        }
    }
    if(k!=0)ans=-1;
    printf("%lld\n",ans);
    return 0;
}