首页 > 代码库 > SPOJ 7258 Lexicographical Substring Search(后缀自动机)
SPOJ 7258 Lexicographical Substring Search(后缀自动机)
【题目链接】 http://www.spoj.com/problems/SUBLEX/
【题目大意】
给出一个字符串,求其字典序排名第k的子串
【题解】
求出sam上每个节点被经过的次数,然后采用权值线段树上查找第k大数类似的方法,
每次确定查找范围,进入相应的子树,同时输出路径上的点即可。
【代码】
#include <cstdio>#include <cstring>#include <algorithm> using namespace std;const int N=500005;char s[N];struct sam{ int p,q,np,nq,cnt,last,a[N][26],l[N],f[N]; sam(){cnt=0;last=++cnt;} void extend(int c){ p=last;np=last=++cnt;l[np]=l[p]+1; while(!a[p][c]&&p)a[p][c]=np,p=f[p]; if(!p)f[np]=1; else{ q=a[p][c]; if(l[p]+1==l[q])f[np]=q; else{ nq=++cnt;l[nq]=l[p]+1; memcpy(a[nq],a[q],sizeof(a[q])); f[nq]=f[q]; f[np]=f[q]=nq; while(a[p][c]==q)a[p][c]=nq,p=f[p]; } } } int b[N],x[N],mx[N]; void build(){ scanf("%s",s+1); int len=strlen(s+1); for(int i=1;i<=len;i++)extend(s[i]-‘a‘); for(int i=1;i<=cnt;i++)b[l[i]]++; for(int i=1;i<=len;i++)b[i]+=b[i-1]; for(int i=1;i<=cnt;i++)x[b[l[i]]--]=i; for(int i=cnt;i;i--){ mx[x[i]]=1; for(int j=0;j<26;j++)mx[x[i]]+=mx[a[x[i]][j]]; } } void query(int k){ p=1; while(k){ for(int i=0;i<26;i++)if(a[p][i]){ if(mx[a[p][i]]>=k){ putchar(‘a‘+i); p=a[p][i]; --k;break; }else k-=mx[a[p][i]]; } }puts(""); }}sam;int main(){ sam.build(); int Q,x; scanf("%d",&Q); while(Q--){ scanf("%d",&x); sam.query(x); }return 0;}
SPOJ 7258 Lexicographical Substring Search(后缀自动机)
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。