Bzoj3998 [TJOI2015]弦论

Time Limit: 10 Sec  Memory Limit: 256 MB
Submit: 2388  Solved: 798

Description

对于一个给定长度为N的字符串，求它的第K小子串是什么。

Input

 第一行是一个仅由小写英文字母构成的字符串S

第二行为两个整数T和K，T为0则表示不同位置的相同子串算作一个。T=1则表示不同位置的相同子串算作多个。K的意义如题所述。

Output

输出仅一行，为一个数字串，为第K小的子串。如果子串数目不足K个，则输出-1

Sample Input

aabc
0 3

Sample Output

aab

HINT

 

 N<=5*10^5

T<2

K<=10^9

 

Source

 

字符串 后缀自动机

T=0 除了根以外的状态都代表1个串

T=1 每个状态fail子树结束结点个数即为串的个数

也就是说如果T等于1，就用每个结点的size更新fa结点，T等于0就不更新

/*by SilverN*/
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<vector>
using namespace std;
const int mxn=1000010;
char s[mxn];
int T;
struct SAM{
    int t[mxn*2][26];
    int fail[mxn*2],l[mxn*2],ct[mxn*2];
    int w[mxn*2],rk[mxn*2],smm[mxn*2];
    int S,last,cnt;
    void init(){S=last=cnt=1;return;}
    void add(int c){
        int p=last,np=++cnt;
        last=np;l[np]=l[p]+1;
//      ct[np]++;
        for(;p && !t[p][c];p=fail[p])t[p][c]=np;
        if(!p){fail[np]=S;}
        else{
            int q=t[p][c];
            if(l[q]==l[p]+1)fail[np]=q;
            else{
                int nq=++cnt;l[nq]=l[p]+1;
                memcpy(t[nq],t[q],sizeof t[q]);
                fail[nq]=fail[q];
                fail[q]=fail[np]=nq;
                for(;p && t[p][c]==q;p=fail[p])t[p][c]=nq;
            }
        }
        return;
    }
    void st(){
        int i,j;int len=strlen(s+1),now=S;
        for(i=1;i<=cnt;i++)w[l[i]]++;
        for(i=1;i<=len;i++)w[i]+=w[i-1];
        for(i=cnt;i;i--)rk[w[l[i]]--]=i;
        for(i=1;i<=len;i++){
            now=t[now][s[i]-'a'];
            ct[now]++;
        }
        for(i=cnt;i;i--){
            int to=rk[i];
            if(T==1)ct[fail[to]]+=ct[to];
            else ct[to]=1;
        }
        ct[1]=0;
        for(i=cnt;i;i--){
            smm[rk[i]]=ct[rk[i]];
            for(j=0;j<26;j++){
                smm[rk[i]]+=smm[t[rk[i]][j]];
            }
        }
        return;
    }
    void solve(int u,int k){
        if(k<=ct[u])return;
//      printf("u:%d k:%d\n",u,k);
        k-=ct[u];
        for(int j=0;j<26;j++){
            if(t[u][j]){
                int v=t[u][j];
                if(k<=smm[v]){
                    printf("%c",j+'a');
                    solve(v,k);
                    return;
                }
                else k-=smm[v];
            }
        }
        return;
    }
}sa;
int n,k;
int main(){
    int i,j;
    scanf("%s",s+1);
    n=strlen(s+1);
    scanf("%d%d",&T,&k);
    sa.init();
    for(i=1;i<=n;i++) sa.add(s[i]-'a');
    sa.st();
//  printf("%d\n",sa.smm[1]);
    if(sa.smm[1]<k)printf("-1\n");
    else sa.solve(1,k);
    return 0;
}