BZOJ2160拉拉队排练（回文树）

Posted 2020-10-25 小蒟蒻yyb的博客

【BZOJ2160】拉拉队排练（回文树）

题面

BZOJ

题解

看着题目，
直接构建回文树
求出每个回文串的出现次数
直接按照长度\(sort\)一下就行了
然后快速幂算一下答案就出来了

这题貌似可以用Manacher做吧
求出以每个字符为中心的回文串的最大长度
然后搞一下前缀和就行了

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define ll long long
#define RG register
#define MAX 1200000
#define MOD 19930726
int size[MAX];
char ch[MAX];
int n,ans=1;
ll K,sum;
pair<int,int> S[MAX];
int fpow(int a,int b)
{
    int s=1;
    while(b){if(b&1)s=1ll*s*a%MOD;a=1ll*a*a%MOD;b>>=1;}
    return s;
}
struct PT
{
    struct Node
    {
        int son[26];
        int ff,len;
    }t[MAX];
    int last,tot;
    void init()
    {
        t[tot=1].len=-1;
        t[1].ff=t[0].ff=1;
    }
    void extend(int c,int n,char *s)
    {
        int p=last;
        while(s[n-t[p].len-1]!=s[n])p=t[p].ff;
        if(!t[p].son[c])
        {
            int v=++tot,k=t[p].ff;
            while(s[n-t[k].len-1]!=s[n])k=t[k].ff;
            t[v].len=t[p].len+2;
            t[v].ff=t[k].son[c];
            t[p].son[c]=v;
        }
        size[last=t[p].son[c]]++;
    }
    void Calc()
    {
        for(int i=tot;i;--i)size[t[i].ff]+=size[i];
    }
}PT;
int main()
{
    scanf("%d %lld",&n,&K);
    scanf("%s",ch+1);
    PT.init();
    for(int i=1;i<=n;++i)PT.extend(ch[i]-97,i,ch);
    PT.Calc();
    for(int i=2;i<=PT.tot;++i)S[i-1]=make_pair(-PT.t[i].len,size[i]);
    sort(&S[1],&S[PT.tot]);
    for(int i=1;i<PT.tot;++i)
    {
        if(S[i].first%2==0)continue;
        ll pl=min(K,1ll*S[i].second);
        ans=1ll*ans*fpow(-S[i].first,pl)%MOD;
        K-=pl;
        if(!K)break;
    }
    if(!K)printf("%d\n",ans);
    else puts("-1");
    return 0;
}