bzoj3998: [TJOI2015]弦论

Posted 2020-10-29 AKCqhzdy

前几天写的题，今天补写个blog

T的情况get right集合的时候特判一下。

因为SAM就是有序的，所以可以dfs求解。

把parent树构建出来，sum表示当前子树的right和，搜一下就出来了。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;

int len,a[1100000];
struct node
{
    int w[30];
}ch[1100000];int root,last,cnt;
int dep[1100000],parent[1100000];
void add(int k)
{
    int x=a[k];
    int now=++cnt,p=last;
    dep[now]=k;
    
    while(p!=0&&ch[p].w[x]==0)ch[p].w[x]=now,p=parent[p];
    if(p==0)parent[now]=root;
    else
    {
        int pre=ch[p].w[x];
        if(dep[p]+1==dep[pre])parent[now]=pre;
        else
        {
            int npre=++cnt;
            ch[npre]=ch[pre];
            dep[npre]=dep[p]+1;
            parent[npre]=parent[pre];
            
            parent[now]=parent[pre]=npre;
            while(ch[p].w[x]==pre)ch[p].w[x]=npre,p=parent[p];
        }
    }
    last=now;
}
int Rsort[1100000],sa[1100000];
int Right[1100000];
void get_Right(int T)
{
    for(int i=1;i<=cnt;i++)Rsort[dep[i]]++;
    for(int i=1;i<=len;i++)Rsort[i]+=Rsort[i-1];
    for(int i=cnt;i>=1;i--)sa[Rsort[dep[i]]--]=i;
    
    int now=root;
    for(int i=1;i<=len;i++)
    {
        int x=a[i];
        now=ch[now].w[x];
        Right[now]++;
    }
    for(int i=cnt;i>=1;i--)
    {
        if(T==1)
            Right[parent[sa[i]]]+=Right[sa[i]];
        else
            Right[sa[i]]=1;
    }
    Right[root]=0;
}

int T,k;
int sum[1100000];
void dfs(int now)
{
    if(k<=Right[now])return ;
    k-=Right[now];
    for(int x=1;x<=26;x++)
    {
        if(ch[now].w[x]!=0)
        {
            int next=ch[now].w[x];
            if(k>sum[next])k-=sum[next];
            else
            {
                printf("%c",x+‘a‘-1);
                dfs(next);
                return ;
            }
        }
    }
}

char ss[1100000];
int main()
{
    scanf("%s",ss+1);len=strlen(ss+1);
    for(int i=1;i<=len;i++)a[i]=ss[i]-‘a‘+1;
    
    root=last=cnt=1;
    for(int i=1;i<=len;i++)add(i);
    
    scanf("%d%d",&T,&k);
    get_Right(T);
    
    for(int i=cnt;i>=1;i--)
    {
        int now=sa[i];sum[now]=Right[now];
        for(int x=1;x<=26;x++)
            if(ch[now].w[x]!=0)sum[now]+=sum[ch[now].w[x]];
    }
    if(k>sum[root])printf("-1\n");
    else dfs(root);
    return 0;
}