spoj7258 Lexicographical Substring Search(SUBLEX) 后缀自动机

    给一个字符串S，若干个查询，每个查询k输出字典序第k小的S的子串。把字符串一次插入到自动机中，由于在不考虑父边的情况下，可以看做一个有向无环图，那么我们可在在这个图上进行dp，令dp[i]表示从节点i开始有多少个子串，转移dp[i]=dp[j]+1（存在一条边c使得i-->j），然后就是从根节点贪心走一遍就可以得到要求的串了。

   另外吐槽一句SPOJ上的题卡时间卡的太凶残了....

#include <iostream>#include <cstdio>#include <memory.h>#include <algorithm>#include <string>#include <cstring>#include <stack>using namespace std;const int maxn=180010;const int S=26;int next[maxn][26];char ch[maxn][26];char str[maxn>>1];int wtop[maxn];int cnt[maxn];int n,m,q;int len,tot;struct node{    node *par,*go[S];    int val,cnt;}*root,*tail,que[maxn],*top[maxn];char ans[maxn];int c[maxn];inline int idx(char c){    return c-'a';}inline void add(int c,int l){    node *p=tail;    node *np=&que[tot++];    np->val=l;    while (p && p->go[c]==NULL) p->go[c]=np,p=p->par;    if (p==NULL) np->par=root;    else    {        node *q=p->go[c];        if (p->val+1==q->val) np->par=q;        else        {            node *nq=&que[tot++];            *nq=*q;            nq->val=p->val+1;            np->par=q->par=nq;            while (p&&p->go[c]==q) p->go[c]=nq,p=p->par;        }    }    tail=np;}inline void init(){    //memset(que,0,sizeof que);    len=1;    tot=0;    root=tail=&que[tot++];}inline void slove(int k){    int now=0;    int l=0;    while (k)    {       for (int i=0; i<cnt[now]; i++)       {           if (k>que[next[now][i]].cnt)           k-=que[next[now][i]].cnt;           else           {               ans[l++]=ch[now][i];               k--;               now=next[now][i];               break;           }       }    }    ans[l]='\0';    puts(ans);}int main(){    //freopen("a.txt","r",stdin);    //freopen("out.txt","w",stdout);    gets(str);    init();    int l=strlen(str);    for (int i=0; i<l; i++)    add(idx(str[i]),len++);   // memset(c,0,sizeof c);   // memset(wtop,0,sizeof wtop);   // memset(cnt,0,sizeof cnt);    for (int i=0; i<tot; i++)    c[que[i].val]++;    for (int i=1; i<len; i++)    c[i]+=c[i-1];    for (int i=0; i<tot; i++)    {        top[--c[que[i].val]]=&que[i];        wtop[c[que[i].val]]=i;    }    for (int i=0; i<tot; i++)    que[i].cnt=1;    for (int i=tot-1; i>=0; i--)    {        node *p=top[i];        for (int j=0; j<S; j++)        {            if (p->go[j])            {                int u=p-que,v=p->go[j]-que;                next[u][cnt[u]]=v;                ch[u][cnt[u]++]=j+'a';                p->cnt+=p->go[j]->cnt;            }        }    }    scanf("%d",&q);    while (q--)    {        scanf("%d",&m);        slove(m);    }    return 0;}