求第k小子串。如果t=0,直接拿sa线性搞即可。如果t=1,我复杂度就炸了。。。然而本机和luogu都跑了过去(可见数据之水,逃)。正解大概是SAM,待更新。
3.24upd:蒟蒻我终于看懂了SAM的板子orz
顾名思义,SAM就是能识别所有后缀的自动机。我们有一个朴素的构建SAM的想法:把每一个后缀都插到Trie里去!但是这样是O(n^2)的,无论时间还是空间都会炸。
怎么办呢?我们要压缩状态,一些子串是可以压在一起的,具体怎么搞可以见cljWC2012讲稿:portal。Candy?的总结也蛮不错的:portal
我们不仅要搞出SAM,还要搞出Parent Tree,其实很多时候我们都是在利用Parent Tree的性质。注意节点是O(2n)的。
把所有节点按mx排序(利用基数排序可以做到O(n)),可以得到Parent Tree及SAM的拓扑序,我们就可以倒着计算出每个节点的|Right|了,或者dp搞些什么事情。总的复杂度都是O(n)的。
对于此题,我们还要处理出sum[p]数组来方便的查询第K小。
SA
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define N 500010
#define ll long long
#define inf 0x3f3f3f3f
inline int read(){
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
return x*f;
}
int type,n,K,sa[N],rk[N<<1],rk1[N],tmp[N],cnt[N],h[N],m='z'+1;
char s[N];
int main(){
// freopen("string.in","r",stdin);
// freopen("string.out","w",stdout);
scanf("%s",s+1);n=strlen(s+1);type=read();K=read();
for(int i=1;i<=n;++i) cnt[s[i]]=1;
for(int i=1;i<=m;++i) cnt[i]+=cnt[i-1];
for(int i=1;i<=n;++i) rk[i]=cnt[s[i]];
int k=0;
for(int p=1;k!=n;p+=p,m=k){
for(int i=1;i<=m;++i) cnt[i]=0;
for(int i=1;i<=n;++i) cnt[rk[i+p]]++;
for(int i=1;i<=m;++i) cnt[i]+=cnt[i-1];
for(int i=n;i>=1;--i) tmp[cnt[rk[i+p]]--]=i;
for(int i=1;i<=m;++i) cnt[i]=0;
for(int i=1;i<=n;++i) cnt[rk[tmp[i]]]++;
for(int i=1;i<=m;++i) cnt[i]+=cnt[i-1];
for(int i=n;i>=1;--i) sa[cnt[rk[tmp[i]]]--]=tmp[i];
memcpy(rk1,rk,sizeof(rk1));rk[sa[1]]=k=1;
for(int i=2;i<=n;++i){
if(rk1[sa[i]]!=rk1[sa[i-1]]||rk1[sa[i]+p]!=rk1[sa[i-1]+p]) ++k;
rk[sa[i]]=k;
}
}k=0;
for(int i=1;i<=n;++i){
if(rk[i]==1){h[1]=0;continue;}
if(i==1||h[rk[i-1]]<=1) k=0;
if(k) --k;
while(s[i+k]==s[sa[rk[i]-1]+k]) ++k;
h[rk[i]]=k;
}for(int i=1;i<=n;++i){
if(type){
for(int j=h[i]+1;sa[i]+j-1<=n;++j){
int r=i+1;
while(h[r]>=j) ++r;--r;
if(K<=r-i+1){
for(int k=1;k<=j;++k) putchar(s[sa[i]+k-1]);return 0;
}K-=r-i+1;
}continue;
}int cnt=n-(sa[i]+h[i])+1;if(K>cnt){K-=cnt;continue;}
for(int j=1;j<=K+h[i];++j) putchar(s[sa[i]+j-1]);return 0;
}puts("-1");return 0;
}
SAM
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 1000010
inline int read(){
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
return x*f;
}
char s[N];
int n=0,m,son[N][26],par[N],mx[N],val[N],sum[N],last,rt,type,K;
int cnt[N],a[N];//sum[p] 就是到达p节点以后能识别的串的个数 val[p]就是p点代表状态的出现次数
inline void ins(int ch){
int p=last,np=++n;last=np;mx[np]=mx[p]+1;val[np]=1;
for(;p&&!son[p][ch];p=par[p]) son[p][ch]=np;
if(!p){par[np]=rt;return;}
int q=son[p][ch];
if(mx[q]==mx[p]+1){par[np]=q;return;}
int nq=++n;mx[nq]=mx[p]+1;memcpy(son[nq],son[q],sizeof(son[q]));
par[nq]=par[q];par[q]=par[np]=nq;
for(;p&&son[p][ch]==q;p=par[p]) son[p][ch]=nq;
}
void dfs(int x){
if(K<=val[x]) return;
K-=val[x];
for(int j=0;j<26;++j){
int y=son[x][j];
if(K<=sum[y]){putchar(j+'a');dfs(y);return;}
K-=sum[y];
}
}
int main(){
// freopen("a.in","r",stdin);
scanf("%s",s+1);m=strlen(s+1);last=rt=++n;
for(int i=1;i<=m;++i) ins(s[i]-'a');
type=read();K=read();
for(int i=1;i<=n;++i) cnt[mx[i]]++;
for(int i=1;i<=m;++i) cnt[i]+=cnt[i-1];
for(int i=n;i>=1;--i) a[cnt[mx[i]]--]=i;
for(int i=n;i>=1;--i){
int p=a[i];type?val[par[p]]+=val[p]:val[p]=1;
}val[1]=0;
for(int i=n;i>=1;--i){
int p=a[i];sum[p]=val[p];
for(int j=0;j<26;++j) sum[p]+=sum[son[p][j]];
}if(K>sum[rt]) puts("-1");
else dfs(rt);
return 0;
}