SPOJ 7258 Lexicographical Substring Search（后缀自动机）

题目链接：http://www.spoj.com/problems/SUBLEX/

题意：给定一个字符串，每次询问第K小的串。

思路：建立自动机。cnt[p]=sigama(cnt[son[p][i]])+1。

const int KIND=26;

char s[N];

int pre[N],Lth[N],son[N][KIND+1];

i64 cnt[N];

int e,n,last;



void insert()

{

    clr(son,-1); e=0;

    pre[e]=-1; Lth[e]=0; last=e++;

    int i,r,p,q,u,x;

    FOR0(i,n)

    {

        x=s[i]-'a';

        u=last; pre[e]=-1; Lth[e]=i+1; last=p=e++;

        for(;u!=-1&&son[u][x]==-1;u=pre[u]) son[u][x]=p;

        if(u==-1) pre[p]=0;

        else

        {

            q=son[u][x];

            if(Lth[q]==Lth[u]+1) pre[p]=q;

            else

            {

                pre[e]=pre[q]; Lth[e]=Lth[u]+1;

                memcpy(son[e],son[q],sizeof(son[e]));

                r=e++;

                pre[p]=pre[q]=r;

                for(;u!=-1&&son[u][x]==q;u=pre[u]) son[u][x]=r;

            }

        }

    }

}



int d[N],Q[N];

char transChar[N];



void init()

{

    int i,j,k,p;

    FOR0(i,n+1) d[i]=0;

    FOR0(i,e) d[Lth[i]]++;

    FOR1(i,n) d[i]+=d[i-1];

    FOR0(i,e) Q[--d[Lth[i]]]=i;

    FORL0(i,e-1)

    {

        p=Q[i]; cnt[p]=1; k=0;

        FOR0(j,KIND) if(son[p][j]!=-1)

        {

            cnt[p]+=cnt[son[p][j]];

            son[p][k++]=son[p][j];

            transChar[son[p][j]]=j+'a';

        }

        son[p][k]=-1;

    }

}



int main()

{

    RD(s); n=strlen(s); insert(); init();

    int Q,len,i,p,q;

    i64 K;

    RD(Q);

    while(Q--)

    {

        RD(K); p=0; len=0;

        while(K)

        {

            i=0;K--;

            while(son[p][i]!=-1)

            {

                q=son[p][i];

                if(K>=cnt[q]) K-=cnt[q];

                else

                {

                    s[len++]=transChar[q];p=q;

                    break;

                }

                i++;

            }

        }

        s[len]=0;

        puts(s);

    }

}