BZOJ2160 拉拉队排练 [manacher][前缀]

F.A.Qs	Home	Discuss	ProblemSet	Status	Ranklist	1 Contest	入门OJ	ModifyUser   Sakura_Lemon	Logout	捐赠本站

Notice:九月份月赛将于本周五晚六点到十一点进行，欢迎大家参加，鸣谢Claris主持，欢迎进群633587999交流！

Problem 2160. – 拉拉队排练

2160: 拉拉队排练

Time Limit: 10 Sec   Memory Limit: 259 MB
Submit: 1822   Solved: 714
[ 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

用manacher算法确定每个位置的回文半径一遍manacher算法, 对于位置i为中心的回文半径R[i], 用dp[i]来表示相邻长度的回文串的数量差分, 因为这里每次更新[1, R[i]]这个区间 + 1, 而只在所有更新完毕之后才查询所以没有必要使用树状数组, 直接根据每次更新的时候dp[1]++, dp[R[i] + 1]–, 最后后dp[1~i]的和就是最终ans[i]的值, 即长度为2*i - 1的回文串的数量，快速幂统计。

#include<iostream>  
#include<cstring>
#include<cstdio>
using namespace std; 
#define ll long long  
#define N 1000100  
const long long mod=19930726; 
int n,R[N];
ll K,dp[N],ans[N];
char in[N];
void Manacher(char *s,int *R,int n) {
    register int p=0,mx=0; 
    R[0]=1;
    for(register int i=1;i<=n;i++){
        if(mx>i)R[i]=min(R[2*p-i],mx-i); 
        else R[i]=1; 
        while(s[i-R[i]]==s[i+R[i]])R[i]++; 
        if(i+R[i]>mx)p=i,mx=i+R[i]; 
    }  
    return; 
}
ll quick_pow(ll base,ll pow){
    register ll ret=1; 
    while(pow){
        if(pow&1)ret=(ret*base)%mod; 
        pow>>=1; 
        base=base*base%mod; 
    }  
    return ret; 
}
template<class T>inline void read(T &res){
    static char ch;T flag=1;
    while((ch=getchar())<'0'||ch>'9')if(ch=='-')flag=-1;res=ch-48;
    while((ch=getchar())>='0'&&ch<='9')res=res*10+ch-48;res*=flag;
}
inline void reads(char *s){
    static char ch;int len=0;
    while((ch=getchar())<'a'||ch>'z');s[++len]=ch;
    while((ch=getchar())>='a'&&ch<='z')s[++len]=ch;
}
int main(){  
    read(n),read(K);
    reads(in);
    in[0]='+';in[n+1]='-';in[n+2]='\0'; 
    Manacher(in,R,n); 
    memset(dp,0,sizeof(dp)); 
    for(int i=1;i<=n;i++){  
        dp[1]++,dp[R[i]+1]--; 
    }  
    memset(ans,0,sizeof(ans)); 
    int maxlen=0; 
    for(register int i=1;i<=(n+1)>>1;i++){  
        ans[i]=ans[i-1]+dp[i]; 
        if(ans[i] > 0)maxlen=i; 
    }
    register ll result=1; 
    register int r=maxlen; 
    while(r&&K>0) 
        result=(result*quick_pow((ll)(2*r-1),min(ans[r],K)))% mod,K-=ans[r],r--; 
    if(K>0)printf("-1\n"); 
    else printf("%lld\n",result);  
    return 0; 
}