2018.12.22 spoj7258 Lexicographical Substring Search（后缀自动机）

传送门

samsamsam基础题。

题意简述：给出一个串，询问第kkk大的本质不同的串。

---

然而这就是弦论的简化版。

我们把samsamsam建出来然后贪心选择就行了。

代码：

#include<bits/stdc++.h>
#define ri register int
using namespace std;
const int N=2e5+5;
int n,q,K;
char s[N];
struct SAM{
	int last,tot,len[N],link[N],son[N][26],rk[N],cnt[N],sum[N];
	SAM(){last=tot=1;}
	inline void expand(int x){
		int p=last,np=++tot;
		len[last=np]=len[p]+1;
		while(p&&!son[p][x])son[p][x]=np,p=link[p];
		if(!p){link[np]=1;return;}
		int q=son[p][x],nq;
		if(len[q]==len[p]+1){link[np]=q;return;}
		len[nq=++tot]=len[p]+1,memcpy(son[nq],son[q],sizeof(son[q])),link[nq]=link[q],link[q]=link[np]=nq;
		while(p&&son[p][x]==q)son[p][x]=nq,p=link[p];
	}
	inline void topsort(){
		for(ri i=1;i<=tot;++i)++cnt[len[i]];
		for(ri i=1;i<=tot;++i)cnt[i]+=cnt[i-1];
		for(ri i=tot;i;--i)rk[cnt[len[i]]--]=i,sum[i]=1;
		for(ri i=tot;i;--i)for(ri j=0;j<26;++j)if(son[rk[i]][j])sum[rk[i]]+=sum[son[rk[i]][j]];
	}
	inline void query(){
		int p=1;
		while(K>0){
			for(ri i=0,v;i<26;++i){
				if(!(v=son[p][i]))continue;
				if(sum[v]>=K){--K,putchar('a'+i),p=v;break;}
				else K-=sum[v];
			}
		}
		puts("");
	}
}sam;
int main(){
	scanf("%s",s),n=strlen(s);
	for(ri i=0;i<n;++i)sam.expand(s[i]-'a');
	sam.topsort(),scanf("%d",&q);
	while(q--)scanf("%d",&K),sam.query();
	return 0;
}