您的位置:首页 > 其它

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

2016-05-30 21:44 387 查看
题意:求一个字符串的长度前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
) 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;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: