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

给出一个$n\le 1e6$的字符串，还有一个询问$k$，现在要求前$k$个长度从打到小奇数长度的回文串的长度乘积。不足$k$个就输出$-1$。
直接回文树全部找出来排序，统计个数。个数够的话直接快速幂计算。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const ll INF=LONG_LONG_MAX;
const int N=1e6+7;
const int mod=19930726;
struct Palindromic_Tree {
	int nxt[N][26],fail[N],len[N];
	int num[N],cnt[N],S[N];
	int lst,n,p;
	int newnode(int l) {
		for(int i=0;i<26;i++) nxt[p][i]=0;
		cnt[p]=num[p]=0;
		len[p]=l;
		return p++;
	}
	void init() {
		p=0;
		newnode(0);
		newnode(-1);
		lst=n=0;
		S[0]=-1;
		fail[0]=1;
	}
	int getfail(int x) {
		while(S[n-len[x]-1]!=S[n]) x=fail[x];
		return x;
	}
	void add(int c) {
		c-='a';
		S[++n]=c;
		int cur=getfail(lst);
		if(!nxt[cur][c]) {
			int now=newnode(len[cur]+2);
			fail[now]=nxt[getfail(fail[cur])][c];
			nxt[cur][c]=now;
			num[now]=num[fail[now]]+1;
		}
		lst=nxt[cur][c];
		cnt[lst]++;
	}
	void count() {
		for(int i=p-1;i>=0;i--) 
			cnt[fail[i]]+=cnt[i]; 
	}
}pam;
char str[N]; 
struct Point {
	int cnt,len;
	bool operator <(const Point &rhs) const {
		return len>rhs.len;
	}
}a[N];
int tot=0;
ll fpow(ll x,ll y) {
	ll ans=1;
	while(y) {
		if(y&1) ans=(ans*x)%mod;
		x=(x*x)%mod;
		y>>=1;
	}
	return ans;
} 
int main() {
	int n;ll k; 
	ll sum=0;
	scanf("%d%lld",&n,&k);
	scanf("%s",str+1);
	pam.init();
	for(int i=1;i<=n;i++)
		pam.add(str[i]);
	pam.count();
	for(int i=2;i<pam.p;i++) {
		if(pam.len[i]&1) {
			a[++tot]={pam.cnt[i],pam.len[i]};
			sum+=pam.cnt[i];
		}
	}
	if(sum<k) {
		puts("-1");
	}
	else {
		ll ans=1;
		sort(a+1,a+1+tot);
		for(int i=1;i<=tot;i++) {
			if(k>=a[i].cnt) {
				ans=(ans*fpow(a[i].len,a[i].cnt))%mod;
				k-=a[i].cnt;	
			}
			else {
				ans=(ans*fpow(a[i].len,k))%mod;
				k=0;
			}
			if(k==0) break;
		} 
		printf("%lld\n",ans);
	}
	return 0;
}