【bzoj3998】[TJOI2015]弦论 后缀自动机

非常好的一道裸题

T=1

每个节点的出现次数=它的right集合大小

初始时，每加入一个节点，它对应的right集合大小就为1

T=0

每个节点算一次，即把right集合大小强制变为1

对深度计数排序后，节点i对fa[i]产生贡献

计算出size[i]表示包含节点i在内，所有的能转移到的子串的出现次数之和

注意：不能按bfs序处理size，因为计算size时，还没计算完cnt，类似拓扑图的bfs和拓扑排序的区别

即后缀自动机的节点按mx排序后，是拓扑序

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>
#define maxn 500010

using namespace std;

int ch[maxn*2][26],mx[maxn*2],size[maxn*2],fa[maxn*2],cnt[maxn*2],q[maxn*2],v[maxn*2];
int n,m,tot=1,T,last=1;
bool vis[2*maxn];
char s[maxn];
long long k;

void insert(int x)
{
	int p=last,np=last=++tot;
	mx[np]=mx[p]+1;cnt[np]=1;
	while (p && !ch[p][x]) ch[p][x]=np,p=fa[p];
	if (!p) fa[np]=1;
	else
	{
		int q=ch[p][x];
		if (mx[q]==mx[p]+1) fa[np]=q;
		else
		{
			int nq=++tot;mx[nq]=mx[p]+1;
			memcpy(ch[nq],ch[q],sizeof(ch[q]));
			fa[nq]=fa[q];
			fa[np]=fa[q]=nq;
			while (p && ch[p][x]==q) ch[p][x]=nq,p=fa[p];
		}
	}
}

void bfs()
{
	for (int i=1;i<=tot;i++) v[mx[i]]++;
	for (int i=1;i<=n;i++) v[i]+=v[i-1];
	for (int i=tot;i>=1;i--) q[v[mx[i]]--]=i;
	for (int i=tot;i>=1;i--)
	{
		int x=q[i];
		if (T==0) cnt[x]=1; else cnt[fa[x]]+=cnt[x];
		if (x==1) cnt[x]=0;
		size[x]=cnt[x];
		for (int j=0;j<26;j++) if (ch[x][j]) size[x]+=size[ch[x][j]];
	}
}

void dfs(int x,long long k)
{
	if (k<=cnt[x]) return;
	k-=cnt[x];
	for (int i=0;i<26;i++)
	  if (ch[x][i])
	  {
	  	if (k<=size[ch[x][i]])
	  	{
	  		printf("%c",i+'a');
	  		dfs(ch[x][i],k);
	  		return;
	  	}
		k-=size[ch[x][i]];
	  }
}

int main()
{
	scanf("%s",s+1);
	n=strlen(s+1);
	for (int i=1;i<=n;i++) insert(s[i]-'a');
	scanf("%d%lld",&T,&k);
	bfs();
	if (k>size[1]) printf("-1\n");
	else dfs(1,k);
	return 0;
}