[BZOJ]2160 拉拉队排练 Manacher+快速幂

2160: 拉拉队排练

Time Limit: 10 Sec   Memory Limit: 259 MB
Submit: 1659   Solved: 658
[ Submit][ Status][ Discuss]

Description

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

Input

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

Output

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

Sample Input

5 3
ababa

Sample Output

45
【样例说明】
和谐小群体女生所拿牌子上写的字母从左到右按照女生个数降序排序后为ababa, aba, aba, bab, a, a, a, b, b，前三个长度的乘积为。

HINT

总共20个测试点，数据范围满足： 

Source

[ Submit][ Status][ Discuss] 
HOME   Back

---

题解

        因为Manacher求出的是极大回文子串，那么对于长度为7的极大回文子串，那么这个字串一定包含了对文串长度为5的，长度为4的.这道题由于是求奇数的，那么我们就不用考虑偶数的情况，Manacher也就不用插‘#’号了.Manacher处理完了之后记录每个长度的个数，再从大到小，每个长度i的个数加上i+2的个数，这里有点像前缀和.然后我们根据题意从大到小取k个即可，过程中对于每种长度快速幂求值(每种长度有多个)，两种长度之间直接相乘(用long long就不会爆)，就可以了.

#include<stdio.h>
#include<algorithm>
using namespace std;
typedef long long dnt;
const int mod=19930726;
const int maxn=2000005;
dnt kk,ans,c;
char s[maxn];
int pal[maxn],cnt[maxn],n,maxx;
void manacher(){
    int mx=0,id=0;
    for(int i=1;i<=n;i++){
        if(mx>i) pal[i]=min(mx-i,pal[2*id-i]);
        else pal[i]=1;
        while(s[i-pal[i]]==s[i+pal[i]]) pal[i]++;
        if(i+pal[i]>mx) mx=i+pal[i],id=i;
	}
}
dnt pow(int a,int k){
	dnt base=(dnt)a,anss=1;
	while(k){
      if(k&1) anss=anss*base%mod;
      base=base*base%mod;
      k=k>>1;
	}
    return anss;
}
int main(){
    scanf("%d%lld",&n,&kk);
    scanf("%s",s+1);s[0]='+',s[n+1] = '-';
    manacher();ans=1;
	for(int i=1;i<=n;i++) cnt[pal[i]*2-1]++,maxx=max(pal[i]*2-1,maxx);
	for(int i=maxx-2;i>=0;i=i-2) cnt[i]+=cnt[i+2];
	for(int i=maxx;i>=0&&kk;i=i-2){
	    if(kk<cnt[i]) c=kk,kk=0;
		else kk-=cnt[i],c=cnt[i];
        ans=ans*pow(i,c)%mod;	    
	}
	printf("%lld\n",ans);
}