【spoj SUBLEX】 Lexicographical Substring Search

http://www.spoj.com/problems/SUBLEX/ (题目链接)

题意

  给出一个字符串,询问其中字典序第K小的子串。

Solution

  后缀自动机例题。

  构出后缀自动机以后,对每个节点预处理出从这个节点可以到达多少个不同的子串。然后就是类似于在平衡树上查找一样沿着SAM一路查找下去一路更新答案了。

  代码借鉴的DaD3zZ大神,感觉好优美。

细节

  后缀自动机的节点数是2N的。

代码

// spoj7528
#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<ctime>
#define LL long long
#define inf 1<<30
#define Pi acos(-1.0)
#define free(a) freopen(a".in","r",stdin),freopen(a".out","w",stdout);
using namespace std; const int maxn=1000010;
char s[maxn],ans[maxn];
int n,K; namespace SAM {
int ch[maxn<<1][26],par[maxn<<1],len[maxn<<1],Dargen,last,sz;
int b[maxn],id[maxn<<1],sum[maxn<<1];
void Extend(int c) {
int np=++sz,p=last;last=np;
len[np]=len[p]+1;
for (;p && !ch[p][c];p=par[p]) ch[p][c]=np;
if (!p) par[np]=Dargen;
else {
int q=ch[p][c];
if (len[p]+1==len[q]) par[np]=q;
else {
int nq=++sz;len[nq]=len[p]+1;
memcpy(ch[nq],ch[q],sizeof(ch[nq]));
par[nq]=par[q];
par[np]=par[q]=nq;
for (;p && ch[p][c]==q;p=par[p]) ch[p][c]=nq;
}
}
}
void build() {
Dargen=last=sz=1;
for (int i=1;i<=n;i++) Extend(s[i]-'a');
}
void pre() { //貌似这个东西是精髓?一个O(n)的基数排序之后根据parent树的性质对Right集合的大小和一些其它奇奇怪怪的东西进行O(n)的预处理
for (int i=1;i<=sz;i++) b[len[i]]++;
for (int i=1;i<=n;i++) b[i]+=b[i-1];
for (int i=1;i<=sz;i++) id[b[len[i]]--]=i;
for (int i=sz,S=0;i>=1;i--,S=0) { //sum记录当前节点不同子串个数
for (int j=0;j<26;j++) S+=sum[ch[id[i]][j]];
sum[id[i]]=S+1;
}
}
void query(int K) {
int now=Dargen,tot=0;
while (K) {
for (int i=0;i<26;i++) if (ch[now][i]) {
if (sum[ch[now][i]]>=K) {
ans[++tot]='a'+i,K--,now=ch[now][i];
break;
}
else K-=sum[ch[now][i]];
}
}
ans[++tot]='\0';
}
}
using namespace SAM; int main() {
scanf("%s",s+1);
n=strlen(s+1);
build();
pre();
int q;scanf("%d",&q);
while (q--) {
scanf("%d",&K);
query(K);
puts(ans+1);
}
return 0;
}
上一篇:Linux下生产者与消费者的线程实现


下一篇:Webpack+Vue+ES6 前端组件化开发mobile-multi-page应用实战总结