当前位置: 代码迷 >> 综合 >> SPOJ-SUBLEX Lexicographical Substring Search(字典序第k小子串,后缀自动机+预处理)
  详细解决方案

SPOJ-SUBLEX Lexicographical Substring Search(字典序第k小子串,后缀自动机+预处理)

热度:87   发布时间:2024-02-12 20:17:41

题意:

给定一个字符串S,求S本质不同排名第k小的子串

数据范围:|S|<=9e4

解法:

对S建立后缀自动机,然后预处理每条路径后面有多少种本质不同的子串,
然后就像权值线段树找第k大那样做就行了。

code:

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxm=2e6+5;
char s[maxm];
int n;
struct SAM{int ch[maxm][26];int fa[maxm],l[maxm];//l[]是等价类的最长字符串长度lenint last=1,tot=1;//tot是节点数量void add(int c){int p=last,np=++tot;last=np;l[np]=l[p]+1;for(;p&&!ch[p][c];p=fa[p])ch[p][c]=np;if(!p)fa[np]=1;else{int q=ch[p][c];if(l[p]+1==l[q])fa[np]=q;else{int nq=++tot;l[nq]=l[p]+1;memcpy(ch[nq],ch[q],sizeof ch[q]);fa[nq]=fa[q];fa[q]=fa[np]=nq;for(;ch[p][c]==q;p=fa[p])ch[p][c]=nq;}}}//本题int cnt[maxm];int idx[maxm];int sz[maxm];void cal_sz(){for(int i=1;i<=tot;i++)sz[i]=1;//基数排序for(int i=1;i<=tot;i++)cnt[l[i]]++;for(int i=1;i<=tot;i++)cnt[i]+=cnt[i-1];for(int i=1;i<=tot;i++)idx[cnt[l[i]]--]=i;//for(int i=tot;i>=1;i--){for(int j=0;j<26;j++){sz[idx[i]]+=sz[ch[idx[i]][j]];}}}void solve(int k){int x=1;while(k){for(int j=0;j<26;j++){if(ch[x][j]){if(sz[ch[x][j]]>=k){printf("%c",'a'+j);x=ch[x][j];k--;break;}else{k-=sz[ch[x][j]];}}}}puts("");}//
}S;
signed main(){//scanf("%s",s+1);n=strlen(s+1);for(int i=1;i<=n;i++){S.add(s[i]-'a');}S.cal_sz();//int q;scanf("%d",&q);while(q--){int k;scanf("%d",&k);S.solve(k);}return 0;
}

  相关解决方案