【SPOJ7258】Lexicographical Substring Search-后缀自动机+拓补序递推

测试地址：Lexicographical Substring Search

题目大意：给定一个字符串，有Q个询问，每个询问字符串中字典序第Ki小的本质不同的子串。

做法：这几天学习了后缀自动机（Suffix Automaton，SAM），个人感觉除了clj的原论文，写得最好的就是这个，我就是看着这个学会后缀自动机的。里面最后讲的一道例题就是这道题，首先对字符串构建后缀自动机，然后对于每个点维护一个值s，指从这个点出发最多能找到的子串数，显然我们可以按原图的反向拓补序依次求出每一个点的s，所以我们可以用类似记忆化搜索的方法递推，搜索到没算过的点继续往下算，算过的就直接累加，当然从一个点出发能找到的子串也包括这个点自身，所以回来时记得+1。然后对于每个询问，从起始节点开始，按照字典序从小到大扫过每个儿子，如果K大于当前儿子能找到的子串数，则将K减去这个数，然后找下一个儿子，直到找到一个儿子使得从它出发能找到的子串数≥K，就输出这个儿子的字母，然后从这个儿子往下寻找。因为从一个点出发能找到的子串也包括这个点自身，所以每走过一个点，都要把K减去1，当K等于0的时候就结束。这样，总的时间复杂度就是O(NQ)，可以通过。

我犯二的地方：第一次写想错了，把最后一个插入的点当成最后一个可以接受后缀的点......显然错了，要分清楚这两个概念。

以下是本人代码：

#include <cstdio>#include <cstdlib>#include <cstring>#include <iostream>#include <algorithm>using namespace std;char s[100010];int n,q,tot=0,in[200010]={0},last=0;struct SAMnode{  int pre,ch[30],step,s;}nd[200010];bool vis[200010]={0};void extend(int c){  int p=last,q,np=++tot,nq;  nd[np].step=nd[p].step+1;  for(int i=0;i<=25;i++) nd[np].ch[i]=-1;  while(nd[p].ch[c]==-1&&p!=-1)  {    nd[p].ch[c]=np;p=nd[p].pre;  }  if (p!=-1)  {    q=nd[p].ch[c];if (nd[q].step==nd[p].step+1){  nd[np].pre=q;}else{  nq=++tot;  nd[nq].pre=nd[q].pre;  for(int i=0;i<=25;i++) nd[nq].ch[i]=nd[q].ch[i];  nd[nq].step=nd[p].step+1;  nd[nq].s=1;  nd[q].pre=nq,nd[np].pre=nq;  while(p!=-1&&nd[p].ch[c]==q)  {    nd[p].ch[c]=nq;p=nd[p].pre;  }}  }  else nd[np].pre=0;  nd[np].s=1;  last=np;}void calc_s(int v){  for(int i=0;i<=25;i++)    if (nd[v].ch[i]!=-1){  if (!vis[nd[v].ch[i]]) calc_s(nd[v].ch[i]);  nd[v].s+=nd[nd[v].ch[i]].s;}  vis[v]=1;}void build(){  nd[0].pre=-1;nd[0].s=0;nd[0].step=0;  for(int i=0;i<=25;i++) nd[0].ch[i]=-1;  for(int i=0;i<n;i++) extend(s[i]-'a');  for(int i=0;i<=tot;i++)    if (nd[i].step==n) {vis[i]=1;break;}  calc_s(0);}void output(int a){  int v=0;  while(a)  {    for(int i=0;i<=25;i++)  if (nd[v].ch[i]!=-1)  {    if (a>nd[nd[v].ch[i]].s) a-=nd[nd[v].ch[i]].s;else {v=nd[v].ch[i];printf("%c",i+'a');break;}  }a--;  }  if (q) printf("\n");}int main(){  scanf("%s",s);  n=strlen(s);  build();    scanf("%d",&q);  while(q--)  {    int a;scanf("%d",&a);output(a);  }    return 0;}