【BZOJ3998】弦论 【后缀自动机】

题意

  给定一个长度为n的字符串，求他的第k小子串是什么。

分析

T=0的时候，这个题跟SPOJ-SUBLEX的做法一样，当T=1的时候，不同位置的子串算多个，那么初始化的时候d[u]=cnt[u]，没走一个字符不是k-1而是k-cnt[u]。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;
const int maxn=1e6+5;
char s[maxn];
int n,T,k;

struct state{
    int len,link;
    int next[26];
}st[2*maxn];
int last,cur,sz;
int cnt[2*maxn],d[2*maxn],c[2*maxn];
void init(){
    sz=1;
    last=cur=0;
    st[0].link=-1;
    st[0].len=0;
//    memset(st[0].next,-1,sizeof(st[0].next));
}

void build_sam(int c){
    cur=sz++;
    st[cur].len=st[last].len+1;
    cnt[cur]=1;
//    memset(st[cur].next,-1,sizeof(st[cur].next));
    int p;
    for(p=last;p!=-1&&st[p].next[c]==0;p=st[p].link){
        st[p].next[c]=cur;
    }
    if(p==-1)
        st[cur].link=0;
    else{
        int q=st[p].next[c];
        if(st[q].len==st[p].len+1)
            st[cur].link=q;
        else{
            int clone=sz++;
            st[clone].len=st[p].len+1;
            st[clone].link=st[q].link;
            for(int i=0;i<26;i++)
                st[clone].next[i]=st[q].next[i];
            for(;p!=-1&&st[p].next[c]==q;p=st[p].link){
                st[p].next[c]=clone;
            }
            st[q].link=st[cur].link=clone;
        }
    }
    last=cur;
}
int cmp(int a,int b){
    return st[a].len>st[b].len;
}
void solve(int k,int ty){
    if(d[0]<k){
        printf("-1
");
        return ;
    }
    int u=0;
    while(k){
        for(int i=0;i<26;i++){
            int v=st[u].next[i];
            if(v==0)continue;
            if(k>d[v])
                k-=d[v];
            else{
                printf("%c",i+'a');
                u=v;
                if(ty==0)
                    k--;
                else
                    k-=cnt[v];
                break;
            }
        }
    }
}
void update(){
    for(int i=1;i<sz;i++){
        int o=c[i];
        for(int j=0;j<26;j++){
            int v=st[o].next[j];
            if(v==0)continue;
            d[o]+=d[v];
        }
    }
    for(int i=0;i<26;i++){
        int v=st[0].next[i];
        if(v==0)continue;
            d[0]+=d[v];
    }
}

int main(){
    scanf("%s",s);
    n=strlen(s);
    init();
    for(int i=0;i<n;i++)
        build_sam(s[i]-'a');
    for(int i=0;i<sz;i++)
        c[i]=i;
    sort(c+1,c+sz,cmp);
    scanf("%d%d",&T,&k);
    if(T==0){
        for(int i=1;i<sz;i++)
            d[i]=1;
        update();
        solve(k,0);
    }else{
        for(int i=1;i<sz;i++){
            int o=c[i];
            if(st[o].link!=-1){
                cnt[st[o].link]+=cnt[o];
            }
        }
        for(int i=1;i<sz;i++)
            d[i]=cnt[i];
        update();
        solve(k,1);
    }
return 0;
}