[BZOJ3998][TJOI2015]弦论(后缀自动机)

211 篇文章 0 订阅
18 篇文章 0 订阅

题目描述

传送门

题解

还是要统计每一个点往后走能有多少个不同的子串。
如果T=0的时候就是主链上的点都是1,然后在parent树上把儿子加到父亲上去
如果T=1的是候则需要计算每一个点right集合的大小
最后在自动机上暴力找第k小的然后输出即可

代码

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define LL long long
#define N 1000005

char s[N];
int n,T,p,np,q,nq,sz,root,last;
int ch[N][30],pre[N],step[N],c[N],pt[N],rt[N];
LL k,sum[N];

void extend()
{
    for (int i=0;i<n;++i)
    {
        int x=s[i]-'a';
        p=last;np=++sz;last=np;
        step[np]=step[p]+1;
        while (p&&!ch[p][x])
        {
            ch[p][x]=np;
            p=pre[p];
        }
        if (!p) pre[np]=root;
        else
        {
            q=ch[p][x];
            if (step[q]==step[p]+1) pre[np]=q;
            else
            {
                nq=++sz;
                step[nq]=step[p]+1;
                memcpy(ch[nq],ch[q],sizeof(ch[q]));
                pre[nq]=pre[q];
                pre[np]=pre[q]=nq;
                while (ch[p][x]==q)
                {
                    ch[p][x]=nq;
                    p=pre[p];
                }
            }
        }
    }
}
void sam()
{
    LL now=0;p=root;
    while (1)
    {
        for (int i=0;i<26;++i)
        {
            if (now+sum[ch[p][i]]<k)
                now+=sum[ch[p][i]];
            else
            {
                putchar('a'+i);
                now+=(LL)rt[ch[p][i]];
                if (now>=k) return;
                p=ch[p][i];
                break;
            }
        }
    }
}
int main()
{
    gets(s);n=strlen(s);
    root=last=++sz;
    extend();
    for (int i=1;i<=sz;++i) ++c[step[i]];
    for (int i=1;i<=n;++i) c[i]+=c[i-1];
    for (int i=sz;i>=1;--i) pt[c[step[i]]--]=i;
    scanf("%d%lld",&T,&k);
    if (!T)
    {
        for (int i=1;i<=sz;++i) rt[i]=1;
    }
    else
    {
        p=root;
        for (int i=0;i<n;++i)
        {
            int x=s[i]-'a';
            p=ch[p][x];
            rt[p]=1;
        }
        for (int i=sz;i>=1;--i)
        {
            p=pt[i];
            rt[pre[p]]+=rt[p];
        }
    }
    for (int i=sz;i>=1;--i)
    {
        p=pt[i];
        if (p!=1) sum[p]=(LL)rt[p];
        for (int j=0;j<26;++j)
            sum[p]+=sum[ch[p][j]];
    }
    if (k>sum[1]) puts("-1");
    else sam();
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值