3998: [TJOI2015]弦论
Time Limit: 10 Sec Memory Limit: 256 MBSubmit: 1481 Solved: 501
[ Submit][ Status][ Discuss]
Description
对于一个给定长度为N的字符串,求它的第K小子串是什么。
Input
第一行是一个仅由小写英文字母构成的字符串S
第二行为两个整数T和K,T为0则表示不同位置的相同子串算作一个。T=1则表示不同位置的相同子串算作多个。K的意义如题所述。
Output
输出仅一行,为一个数字串,为第K小的子串。如果子串数目不足K个,则输出-1
Sample Input
aabc
0 3
0 3
Sample Output
aab
HINT
N<=5*10^5
T<2
K<=10^9
后缀自动机第一题(注释见代码)
#include<iostream>
#include<cstdlib>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define F(i,j,n) for(int i=j;i<=n;i++)
#define D(i,j,n) for(int i=j;i>=n;i--)
#define ll long long
#define maxn 500005
#define maxm 1000005
using namespace std;
int n,t,k;
char s[maxn];
struct sam
{
int last,cnt;
int next[maxm][26],fa[maxm],mx[maxm],val[maxm],sum[maxm],v[maxn],q[maxm];
sam(){last=++cnt;}
void add(int c)
{
int p=last,np=last=++cnt;
mx[np]=mx[p]+1;val[np]=1;
while (!next[p][c]&&p) next[p][c]=np,p=fa[p];
if (!p) fa[np]=1;
else
{
int q=next[p][c];
if (mx[p]+1==mx[q]) fa[np]=q;
else
{
int nq=++cnt;mx[nq]=mx[p]+1;
memcpy(next[nq],next[q],sizeof(next[q]));
fa[nq]=fa[q];fa[np]=fa[q]=nq;
while (next[p][c]==q) next[p][c]=nq,p=fa[p];
}
}
}
void pre()
{
F(i,1,cnt) v[mx[i]]++;
F(i,1,n) v[i]+=v[i-1];
D(i,cnt,1) q[v[mx[i]]--]=i;//q数组保存按mx排序的编号
D(i,cnt,1)
{
int tmp=q[i];
if (t==1) val[fa[tmp]]+=val[tmp];
else val[tmp]=1;
}//val表示Right集合的大小
val[1]=0;
D(i,cnt,1)
{
int tmp=q[i];
sum[tmp]=val[tmp];
F(j,0,25) sum[tmp]+=sum[next[tmp][j]];
}//sum表示从该点出发往下走的子串个数
}
void dfs(int x,int k)
{
if (k<=val[x]) return;//注意返回条件
k-=val[x];
F(i,0,25) if (next[x][i])
{
int tmp=next[x][i];
if (k<=sum[tmp])
{
putchar(i+'a');
dfs(tmp,k);
return;//注意这里的return
}
k-=sum[tmp];
}
}
}a;
int main()
{
scanf("%s",s+1);
scanf("%d%d",&t,&k);
n=strlen(s+1);
F(i,1,n) a.add(s[i]-'a');
a.pre();
if (k>a.sum[1]) puts("-1");
else a.dfs(1,k);
return 0;
}