由於SAM能遍歷所有的子串，只要預處理出某個結點的後繼中有多少個不同的子串就可以了。
首先以每個結點為終態算一個子串，所以初始化計數為1。
然後按照拓撲序，用後繼更新pre。類似數DP那種。
SPOJ很卡時，需要各種優化，首先是經典的拓撲。
一次預處理，把後繼全部存好，以為相應的字母。
有個地方需要注意，在查詢的時候，有個k--。因為以當前字母結束也是一個子串，所以需要減掉
[cpp]
#include<iostream>
#include<cstdio>
#include<map>
#include<cstring>
#include<cmath>
#include<vector>
#include<algorithm>
#include<set>
#include<string>
#include<queue>
#define inf 100000005
#define M 40
#define N 200015
#define maxn 300005
#define eps 1e-10
#define zero(a) fabs(a)<eps
#define Min(a,b) ((a)<(b)?(a):(b))
#define Max(a,b) ((a)>(b)?(a):(b))
#define pb(a) push_back(a)
#define mp(a,b) make_pair(a,b)
#define mem(a,b) memset(a,b,sizeof(a))
#define LL unsigned long long
#define MOD 1000000007
#define lson step<<1
#define rson step<<1|1
#define sqr(a) ((double)(a)*(a))
#define Key_value ch[ch[root][1]][0]
#define test puts("OK");
#pragma comment(linker, "/STACK:1024000000,1024000000")
using namespace std;
struct SAM
{
SAM *pre,*son[26];
int len,cnt;
}*root,*tail,que[N],*b[N];
char str[N/2];
int son[N][26],cnt[N];
int tot=0;
int c[N];
char ch[N][26];
void add(int c,int l)
{
SAM *p=tail,*np=&que[tot++];
np->len=l;
while(p&&p->son[c]==NULL) p->son[c]=np,p=p->pre;
if(p==NULL) np->pre=root;
else
{
SAM *q=p->son[c];
if(p->len+1==q->len) np->pre=q;
else
{
SAM *nq=&que[tot++];
*nq=*q;
nq->len=p->len+1;
np->pre=q->pre=nq;
while(p&&p->son[c]==q) p->son[c]=nq,p=p->pre;
}
}
tail=np;
}
void slove(int k)
{
int l=0;
int now=0;
while(k)
{
for(int i=0;i<cnt[now];i++)
{
if(k>que[son[now][i]].cnt)
{
k-=que[son[now][i]].cnt;
}
else
{
str[l++]=ch[now][i];
now=son[now][i];
k--;
break;
}
}
}
str[l]='\0';
puts(str);
}
int main()
{
root=tail=&que[tot++];
scanf("%s",str);
int l=strlen(str);
for(int i=0;str[i];i++) add(str[i]-'a',i+1);
for(int i=0;i<tot;i++) c[que[i].len]++;
for(int i=1;i<tot;i++) c[i]+=c[i-1];
for(int i=0;i<tot;i++) b[--c[que[i].len]]=&que[i];
for(int i=0;i<tot;i++) que[i].cnt=1;
for(int i=tot-1;i>=0;i--)
{
SAM *p=b[i];
for(int j=0;j<26;j++)
{
if(p->son[j])
{
int u=p-que,v=p->son[j]-que;
son[u][cnt[u]]=v;
ch[u][cnt[u]++]=j+'a';
p->cnt+=p->son[j]->cnt;
}
}
}
int q,k;
scanf("%d",&q);
while(q--)
{
scanf("%d",&k);
slove(k);
}
return 0;
}
#include<iostream>
#include<cstdio>
#include<map>
#include<cstring>
#include<cmath>
#include<vector>
#include<algorithm>
#include<set>
#include<string>
#include<queue>
#define inf 100000005
#define M 40
#define N 200015
#define maxn 300005
#define eps 1e-10
#define zero(a) fabs(a)<eps
#define Min(a,b) ((a)<(b)?(a):(b))
#define Max(a,b) ((a)>(b)?(a):(b))
#define pb(a) push_back(a)
#define mp(a,b) make_pair(a,b)
#define mem(a,b) memset(a,b,sizeof(a))
#define LL unsigned long long
#define MOD 1000000007
#define lson step<<1
#define rson step<<1|1
#define sqr(a) ((double)(a)*(a))
#define Key_value ch[ch[root][1]][0]
#define test puts("OK");
#pragma comment(linker, "/STACK:1024000000,1024000000")
using namespace std;
struct SAM
{
SAM *pre,*son[26];
int len,cnt;
}*root,*tail,que[N],*b[N];
char str[N/2];
int son[N][26],cnt[N];
int tot=0;
int c[N];
char ch[N][26];
void add(int c,int l)
{
SAM *p=tail,*np=&que[tot++];
np->len=l;
while(p&&p->son[c]==NULL) p->son[c]=np,p=p->pre;
if(p==NULL) np->pre=root;
else
{
SAM *q=p->son[c];
if(p->len+1==q->len) np->pre=q;
else
{
SAM *nq=&que[tot++];
*nq=*q;
nq->len=p->len+1;
np->pre=q->pre=nq;
while(p&&p->son[c]==q) p->son[c]=nq,p=p->pre;
}
}
tail=np;
}
void slove(int k)
{
int l=0;
int now=0;
while(k)
{
for(int i=0;i<cnt[now];i++)
{
if(k>que[son[now][i]].cnt)
{
k-=que[son[now][i]].cnt;
}
else
{
str[l++]=ch[now][i];
now=son[now][i];
k--;
break;
}
}
}
str[l]='\0';
puts(str);
}
int main()
{
root=tail=&que[tot++];
scanf("%s",str);
int l=strlen(str);
for(int i=0;str[i];i++) add(str[i]-'a',i+1);
for(int i=0;i<tot;i++) c[que[i].len]++;
for(int i=1;i<tot;i++) c[i]+=c[i-1];
for(int i=0;i<tot;i++) b[--c[que[i].len]]=&que[i];
for(int i=0;i<tot;i++) que[i].cnt=1;
for(int i=tot-1;i>=0;i--)
{
SAM *p=b[i];
for(int j=0;j<26;j++)
{
if(p->son[j])
{
int u=p-que,v=p->son[j]-que;
son[u][cnt[u]]=v;
ch[u][cnt[u]++]=j+'a';
p->cnt+=p->son[j]->cnt;
}
}
}
int q,k;
scanf("%d",&q);
while(q--)
{
scanf("%d",&k);
slove(k);
}
return 0;
}
