BZOJ 3998: [TJOI2015]弦论

BZOJ 3998: [TJOI2015]弦论

标签（空格分隔）： OI-BZOJ OI-后缀自动机

---

Time Limit: 10 Sec
Memory Limit: 256 MB

---

Description

对于一个给定长度为N的字符串，求它的第K小子串是什么。

Input

第一行是一个仅由小写英文字母构成的字符串S

第二行为两个整数T和K，T为0则表示不同位置的相同子串算作一个。T=1则表示不同位置的相同子串算作多个。K的意义如题所述。
Output

输出仅一行，为一个数字串，为第K小的子串。如果子串数目不足K个，则输出-1

Sample Input

aabc

0 3
Sample Output

aab

HINT

N<=5*10^5

T<2

K<=10^9

---

Solution####

对于T=0 的情况，自动机上每个点的权值为1，表示对自动机上可以到达当前状态的状态贡献为1，sum[i]为状态i出发可以走出多少种不同的字符串。
对于T=1 的情况，求出每个状态right集合大小，新建点right集合大小为1，辅助点为0，sum[i]意义同上。

---

Code####

#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<math.h>
#include<algorithm>
#include<queue>
#include<set>
#include<map>
#include<bitset>
#include<vector>
using namespace std;
#define PA pair<int,int>
int read()
{
 	int s=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){s=(s<<1)+(s<<3)+ch-'0';ch=getchar();}
	return s*f;
}
//smile please

const int N=500005;
struct samm
{
	int last,total,n;
	int L[N*2],ch[N*2][26],fa[N*2];
	int val[N*2],sum[N*2],t[N],h[N*2];
	samm(){total=1,last=1;}
	void insert(int C)
	   {
		int p=last,now=last=++total;
		L[now]=L[p]+1;n=L[now];val[now]=1;
		for(;p&&!ch[p][C];p=fa[p])
		    ch[p][C]=now;
		if(!p)fa[now]=1;
		else
		  if(L[ch[p][C]]==L[p]+1)
		    fa[now]=ch[p][C];
		  else
		    {int ne=++total,Q=ch[p][C];
		     memcpy(ch[ne],ch[Q],sizeof(ch[Q]));
		     fa[ne]=fa[Q];
		     L[ne]=L[p]+1;
		     fa[Q]=fa[now]=ne;
		     for(;p&&ch[p][C]==Q;p=fa[p])
		        ch[p][C]=ne;
		    }
	   }
	void pre(int T)
	   {for(int i=1;i<=total;i++)t[L[i]]++;
	    for(int i=1;i<=n;i++)t[i]+=t[i-1];
	    for(int i=total;i;i--)h[t[L[i]]--]=i;
	    for(int i=total,x;i;i--)
	       {x=h[i];
	        if(T)val[fa[x]]+=val[x];
	        else val[x]=1;
		   }
		val[1]=0;
		for(int i=total,x;i;i--)
		   {x=h[i];sum[x]=val[x];
		    for(int j=0;j<26;j++)
		        sum[x]+=sum[ch[x][j]];
		   }
	   }
	void dfs(int x,int K)
	   {K-=val[x];
	    if(K<=0)return;
	    for(int i=0,v;i<26;i++)
	        if(v=ch[x][i])
	           {if(K-sum[v]<=0)
	              {putchar(i+'a');
	               dfs(v,K);
	               break;
				  }
			    K-=sum[v];
			   }
	   }
}a;
char z[N];
int n,k,t; 
int main()
{
	scanf("%s",z);
	n=strlen(z);
	t=read(),k=read();
	for(int i=0;i<n;i++)
	   a.insert(z[i]-'a');
	a.pre(t);
	if(a.sum[1]<k)
	  puts("-1");
	else
	  a.dfs(1,k);
	return 0;
}