【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+;
 char s[maxn];
 int n,T,k;

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

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

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