[TJOI2015]弦论 后缀自动机+贪心

[TJOI2015]

• 考虑后缀自动机每个节点的意义以及endpos
• 从后往前考虑从该点出发能到达几个子串
• 最后在后缀自动机上贪心输出即可

#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define per(i,a,b) for(int i=(a);i>=(b);i--)
using namespace std;
const int N=1e6+100;
int tr[N][30],endpos[N],maxn[N],size[N],bin[N],pre[N],A[N],op,k,n,tmp;
string s;
void insert(int c){
	int q,p=tmp,x=++n; tmp=n,maxn[x]=maxn[p]+1,endpos[x]=1;
	while(p&&!tr[p][c])tr[p][c]=x,p=pre[p];
	if(!p)pre[x]=1;
	else if(maxn[p]+1==maxn[q=tr[p][c]])pre[x]=q;
	else{
		int y=++n; maxn[y]=maxn[p]+1;
		memcpy(tr[y],tr[q],sizeof(tr[q]));
		pre[y]=pre[q],pre[x]=pre[q]=y;
		while(tr[p][c]==q)tr[p][c]=y,p=pre[p];
	}
}
void work(int k,int x){
	if(k<=endpos[x])return;
	k-=endpos[x];
	rep(i,0,25){
		if(tr[x][i]){
			if(k>size[tr[x][i]]){k-=size[tr[x][i]];continue;}
			putchar(i+'a');
			work(k,tr[x][i]);
			return;
		}
	}
}
int main()
{
	//freopen("a.in","r",stdin);
	cin>>s; n=tmp=1;
	rep(i,0,s.size()-1)insert(s[i]-'a');
	rep(i,1,n)bin[maxn[i]]++;
	rep(i,1,n)bin[i]+=bin[i-1];
	rep(i,1,n)A[bin[maxn[i]]--]=i;
	per(i,n,1)endpos[pre[A[i]]]+=endpos[A[i]];
	scanf("%d%d",&op,&k);
	rep(i,1,n) op==0?(size[i]=endpos[i]=1):(size[i]=endpos[i]);
	size[1]=endpos[1]=0;
	per(i,n,1){
		int x=A[i];
		rep(c,0,25)if(tr[x][c])size[x]+=size[tr[x][c]];
	}	
	if(size[1]<k)puts("-1");
	else work(k,1);
	return 0;
}