BZOJ 3998: [TJOI2015]弦论 [后缀自动机 DP]

3998: [TJOI2015]弦论

Time Limit: 10 Sec Memory Limit: 256 MB
Submit: 2152 Solved: 716
[Submit][Status][Discuss]

Description

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

Input

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

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

Output

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

T=1和上一题一样
T=2的话，每个状态不是代表一个子串，而是代表|Right|个子串，然后一样了
判断无解也可以直接与sum[root]比较哦

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <map>
using namespace std;
const int N=1e6+5;
inline int read(){
char c=getchar();int x=0,f=1;
while(c<'0'||c>'9'){if(c=='-')f=-1; c=getchar();}
while(c>='0'&&c<='9'){x=x*10+c-'0'; c=getchar();}
return x*f;
}

int n,Type,k;
int c
,a
,d
,sum
;
char s
;
struct State{
int ch[26],par,val;
}t
;
int sz,root,last;
inline int nw(int _){t[++sz].val=_;return sz;}
inline void iniSAM(){sz=0;root=last=nw(0);}
void extend(int c){
int p=last,np=nw(t[p].val+1);d[np]=1;
for(;p&&!t[p].ch[c];p=t[p].par) t[p].ch[c]=np;
if(!p) t[np].par=root;
else{
int q=t[p].ch[c];
if(t[q].val==t[p].val+1) t[np].par=q;
else{
int nq=nw(t[p].val+1);
memcpy(t[nq].ch,t[q].ch,sizeof(t[q].ch));
t[nq].par=t[q].par;
t[q].par=t[np].par=nq;
for(;p&&t[p].ch[c]==q;p=t[p].par) t[p].ch[c]=nq;
}
}
last=np;
}
void RadixSort(){
for(int i=1;i<=sz;i++) c[t[i].val]++;
for(int i=1;i<=n;i++) c[i]+=c[i-1];
for(int i=1;i<=sz;i++) a[c[t[i].val]--]=i;
}
void dp(){
RadixSort();
for(int i=sz;i>=1;i--){
int u=a[i];
if(Type==1) d[t[u].par]+=d[u];
else d[u]=1;
}
for(int i=sz;i>=1;i--){
int u=a[i];
sum[u]=d[u];
for(int j=0;j<26;j++) sum[u]+=sum[t[u].ch[j]];
}
}
int ans
,p;
void kth(int k){
int u=root;
while(k>0){
for(int i=0;i<26;i++) if(t[u].ch[i]){
int v=t[u].ch[i];
if(sum[v]>=k) {ans[++p]=i;k-=d[v];u=v;break;}
else k-=sum[v];
}
}
}
int main(){
freopen("in","r",stdin);
scanf("%s",s+1);
n=strlen(s+1);
iniSAM();
for(int i=1;i<=n;i++) extend(s[i]-'a');
Type=read();k=read();
dp();
if(sum[1]<k) puts("-1");
else{
kth(k);
for(int i=1;i<=p;i++) putchar(ans[i]+'a');
}
}