SPOJ 7258 Lexicographical Substring Search [后缀自动机 DP]

题意：给一个长度不超过90000的串S，每次询问它的所有不同子串中，字典序第K小的，询问不超过500个。

---

第一道自己做的1A的SAM啦啦啦

很简单，建SAM后跑kth就行了

也需要按val基数排序倒着推出来d[s]状态s的后继子串个数

跑kth的时候判断d[v]>=k的时候就跑到v，并且k应该是--哦，因为一个子串过去了

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <map>
using namespace std;
const int N=2e5+;
inline int read(){
    char c=getchar();int x=,f=;
    while(c<''||c>''){if(c=='-')f=-; c=getchar();}
    while(c>=''&&c<=''){x=x*+c-''; c=getchar();}
    return x*f;
}
 
int n,Q,k;
char s[N];
struct State{
    int ch[],par,val;
}t[N];
int sz,root,last;
inline int nw(int _){t[++sz].val=_;return sz;}
inline void iniSAM(){sz=;root=last=nw();}
void extend(int c){
    int p=last,np=nw(t[p].val+);
    for(;p&&!t[p].ch[c];p=t[p].par) t[p].ch[c]=np;
    if(!p) t[np].par=root;
    else{
        int q=t[p].ch[c];
        if(t[q].val==t[p].val+) t[np].par=q;
        else{
            int nq=nw(t[p].val+);
            memcpy(t[nq].ch,t[q].ch,sizeof(t[q].ch));
            t[nq].par=t[q].par;
            t[q].par=t[np].par=nq;
            for(;p&&t[p].ch[c]==q;p=t[p].par) t[p].ch[c]=nq;
        }
    }
    last=np;
}
int c[N],a[N],d[N];
void RadixSort(){
    for(int i=;i<=sz;i++) c[t[i].val]++;
    for(int i=;i<=n;i++) c[i]+=c[i-];
    for(int i=;i<=sz;i++) a[c[t[i].val]--]=i;
}
void dp(){
    for(int i=sz-;i>=;i--){
        int u=a[i];
        d[u]=;
        for(int j=;j<;j++) d[u]+=d[t[u].ch[j]];//,printf("ch %d %d\n",t[u].ch[j],d[t[u].ch[j]]);
        //printf("state %d %d %d %d %d\n",i,u,t[u].val,d[u],sd[u]);
    }
}
void kth(int k){
    int u=root;
    while(k>){//printf("hi %d %d\n",k,u);
        for(int i=;i<;i++) if(t[u].ch[i]){
            int v=t[u].ch[i];//printf("v %d %d %d\n",v,d[v],sd[v]);
            if(d[v]>=k) {putchar('a'+i);k--;u=v;break;}
            else k-=d[v];
        }
    }
    putchar('\n');
}
int main(){
    freopen("in","r",stdin);
    scanf("%s",s+);
    n=strlen(s+);
    iniSAM();
    for(int i=;i<=n;i++) extend(s[i]-'a');
    RadixSort();
    dp();
    Q=read();
    while(Q--){
        k=read();
        kth(k);
    }
}