[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;
}
和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;
}
相关文章推荐
- win32贪吃蛇实现
- 错误汇总
- Git的使用
- 将1、2、3、4组成任意互不相同的三位数输出,每行控制输出3个数。
- android studio创建项目编译出错问题的解决方法
- 合并两个排序的链表18
- 利用代码扫描工具提升Android代码健壮性
- UML
- MySQL常用函数整理
- __declspec 的用法
- 河南第五届省赛 A奇怪的排序 -- (sort+STL)
- Git开发系列(一):基础
- MongoDB学习笔记
- 设计模式23篇:策略模式
- HDU 4746 Mophues【莫比乌斯反演】
- .attr()与.removeAttr()
- HDU 4746 Mophues【莫比乌斯反演】
- 23种设计模式(12):策略模式
- HDU 4746 Mophues【莫比乌斯反演】
- Oracle查询被锁的表及进程的方法