YbtOJ 「字符串算法」第3章 后缀自动机 C. 【问题3】K小子串 题解--zhengjun

思路

对于 $T=0$ 的情况，直接建出后缀自动机，统计一下一个点为开始的字串有多少个，然后从上往下找就可以了。

对于 $T=1$ 的情况，同样建出后缀自动机，统计一下每个字符串的出现次数，和刚刚一样做就好了。

细节不多。

代码

#include<bits/stdc++.h>
using namespace std;typedef long long ll;const int N=1e6+10;
int n,m,tag,ch[N][26],lnk[N],len[N],las,cnt;char c[N];
namespace Solve1{
	int siz[N],vis[N];
	void extend(int c){
		int now=++cnt,p=las,q,ne;len[now]=len[las]+1;while(~p&&!ch[p][c])ch[p][c]=now,p=lnk[p];if(!~p)lnk[now]=0;else{
			q=ch[p][c];if(len[q]==len[p]+1)lnk[now]=q;else{ne=++cnt;memcpy(ch[ne],ch[q],sizeof ch[q]);len[ne]=len[p]+1;
				lnk[ne]=lnk[q];while(~p&&ch[p][c]==q)ch[p][c]=ne,p=lnk[p];lnk[now]=lnk[q]=ne;}
		}las=now;
	}
	void dfs(int now){if(vis[now])return;vis[now]=1;siz[now]=1;
		for(int i=0;i<26;i++)if(ch[now][i])dfs(ch[now][i]),siz[now]+=siz[ch[now][i]];}
	void solve(){
		m++;for(int i=1;i<=n;i++)extend(c[i]-'a');dfs(0);if(m>=siz[0])puts("-1");else{
			int now=0,i;while(1){m--;if(!m)break;for(i=0;i<26;i++)if(ch[now][i]&&siz[ch[now][i]]<m)m-=siz[ch[now][i]];
				else if(ch[now][i])break;now=ch[now][i];putchar(i+'a');}
		}
	}
}
namespace Solve2{
	int siz[N],cur[N],vis[N];ll sum[N];
	void extend(int c){
		int now=++cnt,p=las,q,ne;len[now]=len[las]+1;while(~p&&!ch[p][c])ch[p][c]=now,p=lnk[p];if(!~p)lnk[now]=0;else{
			q=ch[p][c];if(len[q]==len[p]+1)lnk[now]=q;else{ne=++cnt;memcpy(ch[ne],ch[q],sizeof ch[q]);len[ne]=len[p]+1;
				lnk[ne]=lnk[q];while(~p&&ch[p][c]==q)ch[p][c]=ne,p=lnk[p];lnk[now]=lnk[q]=ne;}
		}siz[las=now]=1;
	}
	void dfs(int now){if(vis[now])return;vis[now]=1;sum[now]=siz[now];
		for(int i=0;i<26;i++)if(ch[now][i])dfs(ch[now][i]),sum[now]+=sum[ch[now][i]];}
	void solve(){
		for(int i=1;i<=n;i++)extend(c[i]-'a');for(int i=1;i<=cnt;i++)cur[i]=i;
		sort(cur+1,cur+1+cnt,[](int x,int y){return len[x]>len[y];});
		for(int i=1,now;now=cur[i],i<=cnt;i++)if(lnk[now])siz[lnk[now]]+=siz[now];
		dfs(0);if(m>=sum[0])puts("-1");else{
			int now=0,i;while(1){m-=siz[now];if(m<=0)break;
			for(i=0;i<26;i++)if(ch[now][i]&&sum[ch[now][i]]<m)m-=sum[ch[now][i]];
				else if(ch[now][i])break;now=ch[now][i];putchar(i+'a');}
		}
	}
}
int main(){
	lnk[0]=-1;scanf("%s%d%d",c+1,&tag,&m);n=strlen(c+1);if(!tag)Solve1::solve();else Solve2::solve();return 0;
}