BZOJ2160 拉拉队排练 回文树+快速幂

思路很简单,PAM板子套一下就行。

先把回文树上所有结点状态的回文串长度和数量存在结构体里,排序,从大到小处理即可,

注意快速幂 不然会T。。

//KX
#include <bits/stdc++.h>
using namespace std;
template<typename T>inline void rd(T&x){
    x=0;int f=1;char ch=getchar();
    while(ch<'0' ||ch>'9'){ if(ch=='-')f=-1; ch=getchar(); }
    while(ch>='0' && ch<='9'){ x=x*10+ch-'0'; ch=getchar(); }
    x*=f;
}
typedef long long ll;
typedef double db;
const int M = 1e6+7000;
const int MOD = 19930726;
#define ls (o<<2)
#define rs (o<<1|1)
#define pb push_back
const int N = 27 ;
struct pam{
    int nxt[M][N],fail[M];//在当前状态首尾添加字符的状态,失配跳到的状态
    int len[M],S[M],ppos[M];//状态i表示的回文长度,缓存池,状态对应首次出现的位置
    int cnt[M],num[M];//状态出现次数,以状态末尾结尾但不包含本条路径的数目
    int last,n,p;//上个状态,总长度,当前状态编号
    int newnode(int l){
        memset(nxt[p],0,sizeof nxt[p]);
        cnt[p]=num[p]=0;
        len[p]=l;
        return p++;
    }
    void init(){
        p=0;
        newnode(0);newnode(-1);
        last=n=0;
        S[n]=-1;
        fail[0]=1;
    }
    int get_fail(int x){
        while(S[n-len[x]-1]!=S[n])x=fail[x];return x;
    }
    void add(int c,int pos){
        c-='a';
        S[++n]=c;
        int cur=get_fail(last);
        if(!nxt[cur][c]){
            int now=newnode(len[cur]+2);
            ppos[now]=pos;
            fail[now]=nxt[get_fail(fail[cur])][c];
            nxt[cur][c]=now;
            num[now]=num[fail[now]]+1;
        }last=nxt[cur][c];cnt[last]++;
    }
    void Count(){
        for(int i=p-1;i>=0;--i)cnt[fail[i]]+=cnt[i];
    }
}p;
//p.p-2本质不同回文串的个数 
char s[M];
struct node
{
	ll le,nm;
	bool operator <(const node b)const
	{
		return le>b.le;
	}
}z[M];
ll qpow(ll a,ll b)//a^b;
{
	ll ans=1;
	while(b)
	{
		if(b&1)
			ans=ans*a%MOD;
		a=a*a%MOD;
		b>>=1;
	}
	return ans;
}
int main()
{
/*	freopen("1.in","r",stdin);
	freopen("1.out","w",stdout);*/
	ll len,m;
	rd(len),rd(m);
	scanf("%s",s);
	p.init();
	for(int i=0;i<len;i++)
		p.add(s[i],i);
	p.Count();
	ll sum=0;
	for(int i=2;i<p.p;i++)
	{
		z[i-2].le=p.len[i],z[i-2].nm=p.cnt[i];
		if(z[i-2].le%2==0)z[i-2].le=0; 
		sum+=z[i-2].nm;
	} 
	if(sum<m)
	{
		puts("-1");
		return 0;
	}
	
	sort(z,z+len);
	ll ans=1,pos=0;
/*	for(int i=0;i<len;i++)
	printf("%d  %d\n",z[i].le,z[i].nm);*/
	while(m)
	{
		if(z[pos].nm>m)
		{
			ans=ans*qpow(z[pos].le,m)%MOD;
			m=0;
		}
		else
		{
			ans=ans*qpow(z[pos].le,z[pos].nm)%MOD;
			m-=z[pos].nm;
		}
		pos++;
	}
	printf("%lld\n",ans);
	return 0;
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值