spoj7258:Lexicographical Substring Search 后缀自动机
生活随笔
收集整理的這篇文章主要介紹了
spoj7258:Lexicographical Substring Search 后缀自动机
小編覺得挺不錯的,現(xiàn)在分享給大家,幫大家做個參考.
經(jīng)典題,找不重復(fù)字典序第k小的字符串。
先跑出SAM,然后在SAM上DP。
因?yàn)閺腟AM的起點(diǎn)跑可以跑出所有的子串,所以我們用dpi表示從i點(diǎn)出發(fā)跑出的子串的個數(shù),顯然有
dpi=∑j=azdpchi,j+1
然后每次按照a..z的順序從起點(diǎn)開始dfs,如果沿chi,j的方向走下去的個數(shù)≤k(即dpchi,j≤k),就沿著這個方向往下走,否則令k=k?dpchi,j,繼續(xù)dfs。
如果求重復(fù)字典序第k小的話,我們可以在加入的時候給np記一下數(shù),最后從起點(diǎn)到點(diǎn)i的串出現(xiàn)的次數(shù)就是i在parent樹中子樹的計(jì)數(shù)之和。(其實(shí)就是求出right集合的大小)
#include<iostream> #include<cstdio> #include<cstring> #define N 200005 using namespace std; char s[N]; int last,S,n,cnt,Q; int fa[N],len[N],ch[N][26],f[N],cc[N],q[N]; inline int read() {int a=0,f=1; char c=getchar();while (c<'0'||c>'9') {if (c=='-') f=-1; c=getchar();}while (c>='0'&&c<='9') {a=a*10+c-'0'; c=getchar();}return a*f; } inline void insert(int c) {int p=last,np=last=++cnt;len[np]=len[p]+1;while (p&&!ch[p][c]) ch[p][c]=np,p=fa[p];if (!p) fa[np]=S;else{int q=ch[p][c];if (len[q]==len[p]+1) fa[np]=q;else{int nq=++cnt;memcpy(ch[nq],ch[q],sizeof(ch[q]));len[nq]=len[p]+1;fa[nq]=fa[q]; fa[q]=fa[np]=nq;while (p&&ch[p][c]==q) ch[p][c]=nq,p=fa[p];}} } void query() {int p=1,x=read();while (x){for (int i=0;i<26;i++)if (ch[p][i]){if (f[ch[p][i]]>=x){putchar('a'+i);p=ch[p][i];x--;break;}else x-=f[ch[p][i]];}}puts(""); } int main() {last=S=++cnt;scanf("%s",s+1);n=strlen(s+1);for (int i=1;i<=n;i++)insert(s[i]-'a');for (int i=1;i<=cnt;i++) ++cc[len[i]];for (int i=1;i<=cnt;i++) cc[i]+=cc[i-1];for (int i=1;i<=cnt;i++) q[cc[len[i]]--]=i;for (int i=cnt;i;i--){f[q[i]]=1;for (int j=0;j<26;j++)f[q[i]]+=f[ch[q[i]][j]];}Q=read();while (Q--) query();return 0; }總結(jié)
以上是生活随笔為你收集整理的spoj7258:Lexicographical Substring Search 后缀自动机的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 4位数字排列组合(所有组合与出现一次)
- 下一篇: 排列数字问题