BZOJ_2160_拉拉队排练_manacher

BZOJ_2160_拉拉队排练_manacher

Description

艾利斯顿商学院篮球队要参加一年一度的市篮球比赛了。拉拉队是篮球比赛的一个看点，好的拉拉队往往能帮助球队增加士气，赢得最终的比赛。所以作为拉拉队队长的楚雨荨同学知道，帮助篮球队训练好拉拉队有多么的重要。拉拉队的选拔工作已经结束，在雨荨和校长的挑选下，n位集优秀的身材、舞技于一体的美女从众多报名的女生中脱颖而出。这些女生将随着篮球队的小伙子们一起，和对手抗衡，为艾利斯顿篮球队加油助威。一个阳光明媚的早晨，雨荨带领拉拉队的队员们开始了排练。n个女生从左到右排成一行，每个人手中都举了一个写有26个小写字母中的某一个的牌子，在比赛的时候挥舞，为小伙子们呐喊、加油。雨荨发现，如果连续的一段女生，有奇数个，并且他们手中的牌子所写的字母，从左到右和从右到左读起来一样，那么这一段女生就被称作和谐小群体。现在雨荨想找出所有和谐小群体，并且按照女生的个数降序排序之后，前K个和谐小群体的女生个数的乘积是多少。由于答案可能很大，雨荨只要你告诉她，答案除以19930726的余数是多少就行了。

Input

输入为标准输入。第一行为两个正整数n和K，代表的东西在题目描述中已经叙述。接下来一行为n个字符，代表从左到右女生拿的牌子上写的字母。

Output

输出为标准输出。输出一个整数，代表题目描述中所写的乘积除以19930726的余数，如果总的和谐小群体个数小于K，输出一个整数-1。

Sample Input

5 3
ababa

Sample Output

45

---

一开始看错题了，以为要奇偶分开讨论。。代码就写成那样了。

本题只考虑长度为奇数的回文串。

manacher过程中求极长回文子串然后区间加差分一下。

剩下的就是模拟了。

代码：

#include <cstdio>
#include <string.h>
#include <algorithm>
using namespace std;
typedef long long ll;
#define mod 19930726
#define N 1000050
int c[N],p[N<<1],a[N<<1],n,d[N],e[N];
char w[N<<1];
ll K;
ll qp(ll x,ll y) {
    ll re=1; for(;y;y>>=1ll,x=x*x%mod) if(y&1ll) re=re*x%mod; return re;
}
int main() {
    scanf("%d%lld",&n,&K);
    int i;
    scanf("%s",w+1);
    for(i=1;i<=n;i++) a[2*i-1]='$',a[2*i]=w[i];
    n=n<<1|1; a[n]='$';
    int mx=0,lst;
    for(i=1;i<=n;i++) {
        if(i<=mx) p[i]=min(p[2*lst-i],mx-i+1);
        else p[i]=1;
        while(i-p[i]>=1&&i+p[i]<=n&&a[i-p[i]]==a[i+p[i]]) p[i]++;
        if(mx<i+p[i]-1) {mx=i+p[i]-1; lst=i;}
        int lim=(i+p[i]-2)/2-(i-p[i]+2)/2+1;
        if(i%2==0) {
            d[1]++; d[(lim+1)/2+1]--;
        }else {
            e[1]++; e[lim/2+1]--;
        }
    }
    n>>=1;
    ll sum=0;
    for(c[1]=sum=d[1],i=3;i<=n;i+=2) {
        c[i]=c[i-2]+d[(i+1)>>1]; sum+=c[i];
    }
    for(i=2;i<=n;i+=2) {
        c[i]=c[i-2]+e[i>>1];
    }
    if(sum<K) {
        puts("-1"); return 0;
    }
    ll ans=1;
    i=n; if(n%2==0) i--;
    for(;i>=1;i-=2) {
        // printf("%d\n",c[i]);
        if(K<=c[i]) {
            printf("%lld\n",ans*qp(i,K)%mod); return 0;
        }
        K-=c[i]; ans=ans*qp(i,c[i])%mod;
    }
}