BZOJ - 3998 弦论 (后缀自动机)

 #include<cstdio>
 #include<cstring>
 #include<queue>
 using namespace std;
 typedef long long ll;
 const int N=1e6+,M=;
 char s[N],s2[N];
 int n,f,k;
 struct SAM {
     int go[N][M],pre[N],mxl[N],c[N],ss[N],tot,last,siz[N],sum[N],vis[N],mx;
     void init() {last=tot=; newnode(); pre[]=-;}
     int newnode(int l) {
         int u=tot++;
         memset(go[u],,sizeof go[u]);
         mxl[u]=l,siz[u]=;
         return u;
     }
     void add(int ch) {
         int p=last,np=last=newnode(mxl[p]+);
         siz[np]=;
         for(; ~p&&!go[p][ch]; p=pre[p])go[p][ch]=np;
         if(!~p)pre[np]=;
         else {
             int q=go[p][ch];
             if(mxl[q]==mxl[p]+)pre[np]=q;
             else {
                 int nq=newnode(mxl[p]+);
                 memcpy(go[nq],go[q],sizeof go[nq]);
                 pre[nq]=pre[q],pre[q]=pre[np]=nq;
                 for(; ~p&&go[p][ch]==q; p=pre[p])go[p][ch]=nq;
             }
         }
     }
     void build(char* s,int n) {init(); for(int i=; i<n; ++i)add(s[i]-'a');}
     void toposort() {
         for(int i=; i<tot; ++i)c[i]=;
         for(int i=; i<tot; ++i)++c[mxl[i]];
         for(int i=; i<tot; ++i)c[i]+=c[i-];
         for(int i=; i<tot; ++i)ss[--c[mxl[i]]]=i;
     }
     void getsiz() {
         for(int i=tot-; i>; --i)siz[pre[ss[i]]]+=siz[ss[i]];
         siz[]=;
     }
     void getsum() {
         for(int i=; i<tot; ++i)sum[i]=siz[i];
         for(int i=tot-; i>=; --i)
             for(int j=; j<M; ++j)if(go[ss[i]][j])sum[ss[i]]+=sum[go[ss[i]][j]];
     }
     void qry(int u,int k) {
         int dep;
         for(dep=; k>siz[u]; ++dep) {
             k-=siz[u];
             for(int i=; i<M; ++i) {
                 int v=go[u][i];
                 if(!v)continue;
                 if(k<=sum[v]) {s2[dep]=i+'a',u=v; break;}
                 k-=sum[v];
             }
         }
         s2[dep]=,puts(s2);
     }
     void run() {
         build(s,n);
         toposort();
         if(f)getsiz();
         else {for(int i=; i<tot; ++i)siz[i]=; siz[]=;}
         getsum();
         if(sum[]<k)puts("-1");
         else qry(,k);
     }
 } sam;
 int main() {
     scanf("%s%d%d",s,&f,&k),n=strlen(s);
     sam.run();
     return ;
 }