BZOJ 2160: 拉拉队排练 回文自动机

2160: 拉拉队排练

Time Limit: 10 Sec  Memory Limit: 259 MB
Submit: 2050  Solved: 780
[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个测试点，数据范围满足： 

回文自动机暴力搞过啊。。

话说csdn换编辑器页面了A

看起来还蛮清爽

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<set>
#include<map>
using namespace std;

typedef long long ll;

inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
void print(int x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}

const int N=1001000,mod=19930726;

int n,ans=1;
ll K;

inline int qpow(int x,int y)
{
	int res(1);
	while(y)
	{
		if(y&1) res=1ll*res*x%mod;
		x=1ll*x*x%mod;
		y>>=1;
	}
	return res;
}

struct PAM
{
	int nt[N][26],fail[N],len[N],cnt[N];
	int suff,sz;
	
	void initial()
	{len[1]=-1;len[2]=0;fail[1]=fail[2]=1;suff=sz=2;}
	
	void insert(char *s,int pos)
	{
		int k=s[pos]-'a',cur=suff;
		while(s[pos]!=s[pos-len[cur]-1]) cur=fail[cur];
		if(!nt[cur][k])
		{
			nt[cur][k]=suff=++sz;
			cnt[sz]=1;len[sz]=len[cur]+2;
			if(len[sz]==1){suff=sz;fail[sz]=2;return ;}
			cur=fail[cur];
			while(s[pos]!=s[pos-len[cur]-1]) cur=fail[cur];
			fail[sz]=nt[cur][k];
		}
		else cnt[suff=nt[cur][k]]++;
	}
	
	void pushup()
	{
		register int i;
		for(i=sz;i>2;i--) cnt[fail[i]]+=cnt[i];
	}
	
	struct node
	{
		int val,num;
		friend bool operator <(const node &x,const node &y)
		{return x.val>y.val;}
	}p[N];
	
	void getans()
	{
		pushup();
		int tot(0);
		register int i,tmp;
		for(i=3;i<=sz;++i)
			if(len[i]&1)
				p[++tot]=(node){len[i],cnt[i]};
		sort(p+1,p+1+tot);
		ll now(0);
		for(i=1;i<=tot;++i)
		{
			if(now>=K) break;
			tmp=min(1ll*p[i].num,K-now);
			ans=1ll*ans*qpow(p[i].val,tmp)%mod;
			now+=p[i].num;
		}
		if(now<K) ans=-1;
	}
	
}pam;

char s[N];

int main()
{
	n=read();scanf("%lld",&K);
	scanf("%s",s+1);
	register int i;
	pam.initial();
	for(i=1;i<=n;++i) pam.insert(s,i);
	pam.getans();
	cout<<ans<<endl;
	return 0;
}