spoj7258:Lexicographical Substring Search 后缀自动机

经典题,找不重复字典序第 k 小的字符串。
先跑出SAM,然后在SAM上DP。
因为从SAM的起点跑可以跑出所有的子串,所以我们用dpi表示从 i 点出发跑出的子串的个数,显然有

dpi=j=azdpchi,j+1

然后每次按照 a..z 的顺序从起点开始 dfs ,如果沿 chi,j 的方向走下去的个数 k (即 dpchi,jk ),就沿着这个方向往下走,否则令 k=kdpchi,j ,继续 dfs

如果求重复字典序第 k 小的话,我们可以在加入的时候给np记一下数,最后从起点到点 i 的串出现的次数就是i parent 树中子树的计数之和。(其实就是求出 right 集合的大小)

#include<iostream>
#include<cstdio>
#include<cstring>
#define N 200005
using namespace std;
char s[N];
int last,S,n,cnt,Q;
int fa[N],len[N],ch[N][26],f[N],cc[N],q[N];
inline int read()
{
    int a=0,f=1; char c=getchar();
    while (c<'0'||c>'9') {if (c=='-') f=-1; c=getchar();}
    while (c>='0'&&c<='9') {a=a*10+c-'0'; c=getchar();}
    return a*f;
}
inline void insert(int c)
{
    int p=last,np=last=++cnt;
    len[np]=len[p]+1;
    while (p&&!ch[p][c]) ch[p][c]=np,p=fa[p];
    if (!p) fa[np]=S;
    else
    {
        int q=ch[p][c];
        if (len[q]==len[p]+1) fa[np]=q;
        else
        {
            int nq=++cnt;
            memcpy(ch[nq],ch[q],sizeof(ch[q]));
            len[nq]=len[p]+1;
            fa[nq]=fa[q]; fa[q]=fa[np]=nq;
            while (p&&ch[p][c]==q) ch[p][c]=nq,p=fa[p];
        }
    }
}
void query()
{
    int p=1,x=read();
    while (x)
    {
        for (int i=0;i<26;i++)
            if (ch[p][i])
            {
                if (f[ch[p][i]]>=x)
                {
                    putchar('a'+i);
                    p=ch[p][i];
                    x--;
                    break;
                }
                else x-=f[ch[p][i]];
            }
    }
    puts("");
}
int main()
{
    last=S=++cnt;
    scanf("%s",s+1);
    n=strlen(s+1);
    for (int i=1;i<=n;i++)
        insert(s[i]-'a');
    for (int i=1;i<=cnt;i++) ++cc[len[i]];
    for (int i=1;i<=cnt;i++) cc[i]+=cc[i-1];
    for (int i=1;i<=cnt;i++) q[cc[len[i]]--]=i;
    for (int i=cnt;i;i--)
    {
        f[q[i]]=1;
        for (int j=0;j<26;j++)
            f[q[i]]+=f[ch[q[i]][j]];
    }
    Q=read();
    while (Q--) query();
    return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值