[BZOJ3998][TJOI2015]弦论

bzoj
luogu

题意

求一个串的第k小子串，不同位置上的相同内容的子串有时会视作同一子串，有时视作不同子串。

sol

基数排序处理\(size\)。
“不同位置上的相同内容的子串是否视为相同”决定了一个状态对答案的贡献到底是\(endpos\)集合大小还是\(1\)。
除此之外就是第\(k\)小的查询了。

code

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int N = 1e6+5;
int n,last=1,tot=1,T,tr[N][26],fa[N],len[N],sz[N],t[N],a[N];
char s[N];long long K,sum[N];
void extend(int c)
{
	int v=last,u=++tot;last=u;
	len[u]=len[v]+1;
	while (v&&!tr[v][c]) tr[v][c]=u,v=fa[v];
	if (!v) fa[u]=1;
	else
	{
		int x=tr[v][c];
		if (len[x]==len[v]+1) fa[u]=x;
		else
		{
			int y=++tot;
			memcpy(tr[y],tr[x],sizeof(tr[y]));
			fa[y]=fa[x];fa[x]=fa[u]=y;len[y]=len[v]+1;
			while (v&&tr[v][c]==x) tr[v][c]=y,v=fa[v];
		}
	}
	sz[u]=1;
}
void Query(int u,long long k)
{
	if (k<=sz[u]) return;k-=sz[u];
	for (int j=0;j<26;++j)
	{
		if (k>sum[tr[u][j]]) k-=sum[tr[u][j]];
		else
		{
			putchar(j+'a');
			Query(tr[u][j],k);
			return;
		}
	}
}
int main()
{
	scanf("%s",s+1);n=strlen(s+1);
	scanf("%d%lld",&T,&K);
	for (int i=1;i<=n;++i) extend(s[i]-'a');
	for (int i=1;i<=tot;++i) ++t[len[i]];
	for (int i=1;i<=tot;++i) t[i]+=t[i-1];
	for (int i=1;i<=tot;++i) a[t[len[i]]--]=i;
	for (int i=tot;i;--i) sz[fa[a[i]]]+=sz[a[i]];
	for (int i=1;i<=tot;++i) if (T) sum[i]=sz[i];else sum[i]=sz[i]=1;
	sum[1]=sz[1]=0;//因为1是初始的根节点，本质上并不对应任何子串
	for (int i=tot;i;--i)
		for (int j=0;j<26;++j)
			sum[a[i]]+=sum[tr[a[i]][j]];
	if (sum[1]<K) puts("-1");
	else Query(1,K),puts("");
	return 0;
}