傳送門

解題思路

　　首先建\(sam\)，然后在拓撲序上\(dp\)一下，把每個點的路徑數算出來，然后統計答案時就在自動機上\(dfs\)一下，仿照平衡樹那樣找第\(k\)小。

代碼

#include<iostream> #include<cstdio> #include<cstring> #include<cmath>using namespace std; const int MAXN = 90005;inline int rd(){int x=0,f=1;char ch=getchar();while(!isdigit(ch)) f=ch=='-'?0:1,ch=getchar();while(isdigit(ch)) x=(x<<1)+(x<<3)+ch-'0',ch=getchar();return f?x:-x; }int n,Q,l[MAXN<<1],fa[MAXN<<1],ch[MAXN<<1][27]; int lst,cnt,f[MAXN<<1],c[MAXN<<1],a[MAXN<<1]; char s[MAXN];inline void Insert(int c){int p=lst,np=++cnt;lst=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[q]==l[p]+1) fa[np]=q;else{int nq=++cnt;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;}} } void query(int x){int p=1;while(x){for(int i=1;i<=26;i++)if(ch[p][i]){if(f[ch[p][i]]<x) x-=f[ch[p][i]];else {p=ch[p][i];putchar('a'+i-1);x--;break;} }}putchar('\n'); }int main(){scanf("%s",s+1);n=strlen(s+1);lst=cnt=1;for(int i=1;i<=n;i++) Insert(s[i]-'a'+1);for(int i=1;i<=cnt;i++) c[l[i]]++;for(int i=1;i<=cnt;i++) c[i]+=c[i-1];for(int i=1;i<=cnt;i++) a[c[l[i]]--]=i;for(int i=cnt;i;i--){f[a[i]]=1;for(int j=1;j<=26;j++)f[a[i]]+=f[ch[a[i]][j]];}f[1]--;Q=rd();int x;while(Q--){x=rd();query(x);}return 0; }

轉載于:https://www.cnblogs.com/sdfzsyq/p/10106145.html

總結

以上是生活随笔為你收集整理的SP7258 SUBLEX - Lexicographical Substring Search(后缀自动机)的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。