[BZOJ2160]拉拉队排练(回文树)

题意:求一个字符串的长度前k的奇数长度回文串的长度的乘积模一个数。

和APIO那个回文串的题差不多,可以先manacher搞出所有本质不同的回文串,再用后缀数组查查出现了多少次即可。这里偷懒写的回文树,注意跑完了之后那个cnt需要沿着后缀链接向上传递一下。

注意那个K是10^12不是10^9。。。开的int丢了5分,还好出题人比较良心只有一个点的K超了int。

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#define rep(i,a,b) for(int i=a;i<=b;++i)
#define erp(i,a,b) for(int i=a;i>=b;--i)
#define LL long long
using namespace std;
const int mo = 19930726;
const int MAXN = 1000005;
int N;
LL sum[MAXN], K;

int ksm(int a, int b)
{
	int res = 1;
	for (; b; b>>=1, a=1ll*a*a%mo)
		if(b&1) res = 1ll*res*a%mo;
	return res;
}

int ch[MAXN][26], fa[MAXN];
int cnt[MAXN], len[MAXN];
int last, n, ncnt;
char s[MAXN];
inline int NewNode(int l)
{
	return len[ncnt] = l, ncnt++;
}
void initpam()
{
	NewNode(0), NewNode(-1);
	s[0] = -1, fa[0] = 1;
}
inline int getfail(int x)
{
	while (s[n-len[x]-1] != s[n]) x = fa[x];
	return x;
}

void ins(int c)
{
	s[++n] = c;
	int cur = getfail(last);
	if (!ch[cur][c])
	{
		int now = NewNode(len[cur]+2);
		fa[now] = ch[getfail(fa[cur])][c];
		ch[cur][c] = now;
	}
	last = ch[cur][c];
	cnt[last] ++;
}

void calc()
{
	erp(i, ncnt-1, 0) cnt[fa[i]] += cnt[i];
	for (int i = 0; i<ncnt; ++i)
		if (len[i]>0&&(len[i]&1)) sum[len[i]] += cnt[i];
}

int main()
{
	freopen("rehearse.in","r",stdin);
	freopen("rehearse.out","w",stdout);
	char c;
	scanf("%d %lld\n", &N, &K);
	initpam();
	while (1)
	{
		c = getchar();
		if (c>='a'&&c<='z') ins(c-'a');
		else break;
	}
	calc();
	int ans = 1;
	for (int i = N; i; --i)
	{
		if (!(i&1)) continue;
		if (K<=sum[i]) { ans = 1ll*ans*ksm(i, K)%mo; K = 0; break; }
		else ans = 1ll*ans*ksm(i, sum[i])%mo, K -= sum[i];
	}
	if (K) puts("-1");
	else printf("%d\n", ans);
	return 0;
}


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

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值