BZOJ_3998_[TJOI2015]弦論_後綴自動機
阿新 • • 發佈:2018-06-08
整數 字母 In 英文 out pan str spa test
0 3
首先肯定是要一位一位確定字符,這樣我們就需要兩個標記。
建出後綴自動機。
設siz[i]表示i對應子串在串中出現了多少次,f[i]表示i對應子串有多少種轉移方法。
siz[i]比較好求,如果表示不同位置的相同子串算作一個所有siz都為1,否則siz可以通過後綴樹DP出來。
f[i]用拓撲序DP出來,初始值為siz。
然後按位確定每一位的字符即可。
代碼:
BZOJ_3998_[TJOI2015]弦論_後綴自動機
Description
對於一個給定長度為N的字符串,求它的第K小子串是什麽。
Input
第一行是一個僅由小寫英文字母構成的字符串S
第二行為兩個整數T和K,T為0則表示不同位置的相同子串算作一個。T=1則表示不同位置的相同子串算作多個。K的意義如題所述。Output
輸出僅一行,為一個數字串,為第K小的子串。如果子串數目不足K個,則輸出-1
Sample Input
aabc0 3
Sample Output
aabHINT
N<=5*10^5
T<2 K<=10^9#include <cstdio> #include <string.h> #include <algorithm> using namespace std; #define N 1000050 int ch[N][26],fa[N],dep[N],siz[N],cnt=1,lst=1,f[N]; int ws[N],a[N]; char w[N]; void insert(int x) { int p=lst,np=++cnt,q,nq; lst=np; dep[np]=dep[p]+1; for(;p&&!ch[p][x];p=fa[p]) ch[p][x]=np; if(!p) fa[np]=1; else { q=ch[p][x]; if(dep[q]==dep[p]+1) fa[np]=q; else { fa[nq=++cnt]=fa[q]; dep[nq]=dep[p]+1; memcpy(ch[nq],ch[q],sizeof(ch[q])); fa[q]=fa[np]=nq; for(;p&&ch[p][x]==q;p=fa[p]) ch[p][x]=nq; } } siz[np]=1; } void print() { int i,j; printf("test-------------------------------------------\n"); for(i=1;i<=cnt;i++) { printf("p=%d,siz=%d,dep=%d,fa=%d\n",i,siz[i],dep[i],fa[i]); for(j=0;j<26;j++) { if(ch[i][j]) { printf("ch(%d)(%c)=%d\n",i,j+‘a‘,ch[i][j]); } } } printf("lst=%d\n",lst); } int main() { int T,K; scanf("%s%d%d",w+1,&T,&K); int n=strlen(w+1); int i,j,p; for(i=1;i<=n;i++) insert(w[i]-‘a‘); for(i=1;i<=cnt;i++) ws[dep[i]]++; for(i=1;i<=n;i++) ws[i]+=ws[i-1]; for(i=cnt;i;i--) a[ws[dep[i]]--]=i; if(T) { for(i=cnt;i;i--) { p=a[i]; siz[fa[p]]+=siz[p]; } }else for(i=cnt;i;i--) siz[a[i]]=1; for(siz[1]=0,i=cnt;i;i--) { p=a[i]; for(f[p]=siz[p],j=0;j<26;j++) f[p]+=f[ch[p][j]]; } if(f[1]<K) puts("-1"); else { p=1; while(K>siz[p]) { K-=siz[p]; for(i=0;i<26;i++) { if(K<=f[ch[p][i]]) {printf("%c",i+‘a‘),p=ch[p][i]; break;} else K-=f[ch[p][i]]; } } puts(""); } }
BZOJ_3998_[TJOI2015]弦論_後綴自動機