【GDOI2012】字符串

Description

mmm正在学习字典序。现在老师给她布置了一个作业：给出一个字符串，问该字符串的所有不同的子串中，按字典序排第K的字串。由于众所周知的原因，mmm需要你为她解决这个问题。

Solution

这里与TJOI2015弦论几乎是一样的，比那题还要水很多。
根本就不用担心爆栈什么的。
后缀自动机直接秒过。

Code

#include<iostream> 
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fod(i,a,b) for(i=a;i>=b;i--)
using namespace std;
const int maxn=20007;
int i,j,k,l,n,m,ans;
char s[maxn];
struct node{
    int len,fa,son[26],sum;
}t[maxn*2];
int a[maxn*2],b[maxn*2],c[maxn*2];
int last,num,np,p,q,nq;
void extend(int c){
    np=++num,p=last;
    t[np].len=t[p].len+1;
    while(p&&!t[p].son[c])t[p].son[c]=np,p=t[p].fa;
    if(!p)t[np].fa=1;
    else{
        q=t[p].son[c];
        if(t[p].len+1==t[q].len)t[np].fa=q;
        else{
            nq=++num;
            t[nq]=t[q];
            t[nq].len=t[p].len+1;
            t[q].fa=t[np].fa=nq;
            while(p&&t[p].son[c]==q)t[p].son[c]=nq,p=t[p].fa;
        }
    }
    last=np;
}
void dfs(int x,int k){
    if(k<=c[x])return;
    k-=c[x];
    int j;
    fo(j,0,25){
        if(t[x].son[j]){
            if(k<=t[t[x].son[j]].sum){
                putchar(j+'a');
                dfs(t[x].son[j],k);
                return;
            }
            k-=t[t[x].son[j]].sum;
        }
    }
}
int main(){
    scanf("%s",s+1);
    n=strlen(s+1);
    last=num=1;
    fo(i,1,n)extend(s[i]-'a');
    fo(i,1,num)a[t[i].len]++;
    fo(i,1,n)a[i]+=a[i-1];
    fod(i,num,1)b[a[t[i].len]--]=i;
    fo(i,2,num)c[i]=1;
    fod(i,num,1){
        int u=b[i];t[u].sum=c[u];
        fo(j,0,25){
            t[u].sum+=t[t[u].son[j]].sum;
        }
    }
    scanf("%d",&m);
    fo(i,1,m){
        scanf("%d",&k);
        if(t[1].sum<k)printf("-1\n");
        else {
            dfs(1,k);
            printf("\n");
        }
    }
}