bzoj3998

题意：给出字符串，求其中字典序第k小的子串。

我们设Ans[s]为状态s之后一共有多少子串，这个在建完图之后倒叙更新一遍就可以了。

#include<iostream>
#include<cstring>
#include<cstdlib>
#include<string>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<ctime>
using namespace std;
int n,T,K;
char s[1000010];
struct node{
	int tot,son[1000010][26],sum[1000010],pre[1000010],val[1000010];
	int last,step[1000010],num[1000010],v[1000010],Ans[1000010];
	node () { last=++tot; }
	void insert(int x) {
		int now=last,np=last=++tot;step[np]=step[now]+1;val[np]=1;
		while(son[now][x]==0&&now!=0) son[now][x]=np,now=pre[now];
		if(now==0) pre[np]=1;
		else 
		{
			int q=son[now][x];
			if(step[q]==step[now]+1) pre[np]=q;
			else 
			{
				int newq=++tot;
				step[newq]=step[now]+1;
				for(int i=0;i<26;i++) son[newq][i]=son[q][i];
				pre[newq]=pre[q];
				pre[q]=pre[np]=newq;
				while(son[now][x]==q) son[now][x]=newq,now=pre[now];
			}
		}
	}
	void calc() {
		for(int i=1;i<=tot;i++) num[step[i]]++;
		for(int i=1;i<=n;i++) num[i]+=num[i-1];
		for(int i=tot;i>=1;i--) v[num[step[i]]--]=i;
		for(int i=tot;i>=1;i--) 
		{
			if(T==1) val[pre[v[i]]]+=val[v[i]];
			else val[v[i]]=1;
		}
		val[1]=0;
		for(int i=tot;i>=1;i--)
		{
			Ans[v[i]]=val[v[i]];
			for(int j=0;j<26;j++) 
			    Ans[v[i]]+=Ans[son[v[i]][j]];
		} 
	}
	void dfs(int x,int G) {
		if(val[x]>=G) return ;
		G-=val[x];
		for(int i=0;i<26;i++)
		{
			if(son[x][i]!=0&&G<=Ans[son[x][i]])
			{
				printf("%c",'a'+i);	
				dfs(son[x][i],G);
				return ;		
			}
			G-=Ans[son[x][i]];
		}     
	}
};node SAM;
int main()
{
	scanf("%s",s+1); 
	n=strlen(s+1); 
	scanf("%d%d",&T,&K);
	for(int i=1;i<=n;i++) SAM.insert(s[i]-'a');
	SAM.calc();
	if(K>SAM.Ans[1]) printf("-1");
	else SAM.dfs(1,K);
	return 0;
}