spoj 7258 SUBLEX（SAM，名次）

 

【题目链接】

 

　　http://www.spoj.com/problems/SUBLEX/en/

 

【题意】

 

       给定一个字符串，询问次序为k的子串。

 

【思路】

 

　　SAM，名次

　　建好SAM后求出每个结点根据trans转移向下统计有多少与该结点有相同前缀的子串，每个结点只算一个，因为该结点所代表的字符串集中只可能有一个与上面统计下来的字符串前缀相同，其他的字符串都与该字符串为后缀关系。

　　设sum(p)为与root->p有相同前缀的字符串，设np=trans(p,x)，则有转移式：

　　　　sum(p)=sigma{ sum(np) }

　　找名次为k的子串，只需要按照sum进行转移即可。

 

【代码】

 

#include<cstdio>
#include<cstring>
using namespace std;

const int N = 3*1e5+10;

char s[N];
int root,last,sz,ch[N][26],fa[N],sum[N],l[N];
void add(int x) {
    int c=s[x]-'a';
    int p=last,np=++sz; last=np;
    l[np]=x+1;
    for(;p&&!ch[p][c];p=fa[p]) ch[p][c]=np;
    if(!p) fa[np]=root;
    else {
        int q=ch[p][c];
        if(l[p]+1==l[q]) fa[np]=q;
        else {
            int nq=++sz; l[nq]=l[p]+1;
            memcpy(ch[nq],ch[q],sizeof(ch[q]));
            fa[nq]=fa[q];
            fa[np]=fa[q]=nq;
            for(;p&&q==ch[p][c];p=fa[p]) ch[p][c]=nq;
        }
    }
}

int n,m,b[N],cnt[N];

void get_sum() {
    for(int i=1;i<=sz;i++) cnt[l[i]]++;
    for(int i=1;i<=n;i++) cnt[i]+=cnt[i-1];
    for(int i=1;i<=sz;i++) b[cnt[l[i]]--]=i;
    for(int i=sz;i;i--) {
        int p=b[i]; sum[p]=1;
        for(int j=0;j<26;j++)
            sum[p]+=sum[ch[p][j]];
    }
}

int main() {
    scanf("%s",s);
    n=strlen(s);
    root=last=++sz;
    for(int i=0;i<n;i++) add(i);
    get_sum();
    scanf("%d",&m);
    while(m--) {
        int x,p=root; scanf("%d",&x);
        while(x) {
            for(int c=0;c<26;c++)if(ch[p][c]) {
                if(sum[ch[p][c]]>=x) {
                    putchar('a'+c);
                    x--; p=ch[p][c];
                    break;
                }
                else x-=sum[ch[p][c]];
            }
        }
        puts("");
    }
    return 0;
}