bzoj3998 [TJOI2015]弦论
题目大意：
求第k小的子串。
解题方法：
首先构建后缀自动机，通过拓扑排序求出每个子串的cnt（即该串的出现次数）。然后跑DFS，类似于平衡树求rank为k的数是什么的操作，如果当前指向的儿子（假装他是儿子）的cnt小于k，则k-=cnt，并转向下一个儿子。
代码：
1 #include<cstdio> 2 #include<cstdlib> 3 #include<cstring> 4 #include<algorithm> 5 #include<cmath> 6 #include<iostream> 7 #include<queue> 8 #define ll long long 9 #define del(a,b) memset(a,b,sizeof(a)) 10 using namespace std; 11 const int MAXN=1e6+10; 12 int op; 13 int c[MAXN],a[MAXN]; 14 char s[MAXN]; 15 struct SAM 16 { 17 struct State 18 { 19 int ch[26],fa,val; 20 }t[MAXN]; 21 int n,sz,root,last; 22 int cnt[MAXN],sum[MAXN]; 23 int nw(int x){t[++sz].val=x;return sz;} 24 void init() 25 { 26 sz=0;root=last=nw(0); 27 del(cnt,0);del(sum,0); 28 } 29 void extend(int c) 30 { 31 int p=last,np=nw(t[p].val+1); 32 for(;p&&!t[p].ch[c];p=t[p].fa) t[p].ch[c]=np; 33 if(!p) t[np].fa=root; 34 else 35 { 36 int q=t[p].ch[c]; 37 if(t[q].val==t[p].val+1) t[np].fa=q; 38 else 39 { 40 int nq=nw(t[p].val+1); 41 memcpy(t[nq].ch,t[q].ch,sizeof(t[q].ch)); 42 t[nq].fa=t[q].fa; 43 t[q].fa=t[np].fa=nq; 44 for(;p&&t[p].ch[c]==q;p=t[p].fa) t[p].ch[c]=nq; 45 } 46 } 47 cnt[np]=1;last=np; 48 } 49 void RadixSort() 50 { 51 del(c,0); 52 for(int i=1;i<=sz;i++) c[t[i].val]++; 53 for(int i=1;i<=n;i++) c[i]+=c[i-1]; 54 for(int i=sz;i>=1;i--) a[c[t[i].val]--]=i; 55 } 56 void get_pre() 57 { 58 for(int i=sz;i>=1;i--) 59 if(op==1) cnt[t[a[i]].fa]+=cnt[a[i]]; 60 else cnt[a[i]]=1; 61 cnt[1]=0; 62 for(int i=sz;i>=1;i--) 63 { 64 sum[a[i]]=cnt[a[i]]; 65 for(int j=0;j<26;j++) sum[a[i]]+=sum[t[a[i]].ch[j]]; 66 } 67 } 68 void DFS(int x,int k) 69 { 70 if(k<=cnt[x]) return; 71 k-=cnt[x]; 72 for(int i=0;i<26;i++) 73 if(int to=t[x].ch[i]) 74 { 75 if(k<=sum[to]) 76 { 77 putchar(i+'a'); 78 DFS(to,k);return; 79 } 80 k-=sum[to]; 81 } 82 } 83 }sam; 84 int main() 85 { 86 int k;sam.init(); 87 scanf("%s",s+1); 88 sam.n=strlen(s+1); 89 scanf("%d%d",&op,&k); 90 for(int i=1;i<=sam.n;i++) sam.extend(s[i]-'a'); 91 sam.RadixSort();sam.get_pre(); 92 if(k>sam.sum[1]) puts("-1"); 93 else sam.DFS(1,k); 94 return 0; 95 }
