【BZOJ 3998】 3998: [TJOI2015]弦论 (SAM )

3998: [TJOI2015]弦论

Time Limit: 10 Sec  Memory Limit: 256 MB
Submit: 2627  Solved: 881

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

【分析】

　　建SAM，然后跑。

　　right数组要按照拓扑序来求啊！！！！

　　然后累计儿子的和的时候也要用拓扑序。

　　具体拓扑序：

　　

for(int i=1;i<=tot;i++) v[t[i].step]++;
for(int i=1;i<=tot;i++) v[i]+=v[i-1];
for(int i=tot;i>=1;i--) q[v[t[i].step]--]=i;

　　类似后缀数组那里的了。

　　T=0，就right一开始都为1；T=1，就用right数组。

　　空串算一个串，一开始k++。

　　当然后缀数组也是可以的。

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define Maxn 500010

struct node
{
    int pre,last,son[30],step;
}t[Maxn*2];
int rt[Maxn*2],sm0[2*Maxn],sm1[2*Maxn];
int v[2*Maxn],q[2*Maxn];

struct sam
{
    int last,tot;
    void extend(int k)
    {
        int np=++tot,p=last;
        t[np].step=t[last].step+1;
        rt[np]=1;
        while(p&&!t[p].son[k])
        {
            t[p].son[k]=np;
            p=t[p].pre;
        }
        if(!p) t[np].pre=1;
        else
        {
            int q=t[p].son[k];
            if(t[q].step==t[p].step+1) t[np].pre=q;
            else
            {
                int nq=++tot;
                memcpy(t[nq].son,t[q].son,sizeof(t[nq].son));
                t[nq].step=t[p].step+1;
                t[nq].pre=t[q].pre;
                t[q].pre=t[np].pre=nq;
                while(p&&t[p].son[k]==q)
                {
                    t[p].son[k]=nq;
                    p=t[p].pre;
                }
            }
        }
        last=np;
    }
    void init()
    {
        for(int i=1;i<=tot;i++) v[t[i].step]++;
        for(int i=1;i<=tot;i++) v[i]+=v[i-1];
        for(int i=tot;i>=1;i--) q[v[t[i].step]--]=i;
        
        // for(int i=1;i<=tot;i++) rt[i]=1;
        // for(int i=tot;i>=1;i--) rt[t[i].pre]+=rt[i];
        // rt[1]=1;
        for(int i=tot;i>=1;i--)
        {
            int nw=q[i];
            rt[t[nw].pre]+=rt[nw];
        }rt[1]=1;
        
        for(int i=tot;i>=1;i--)
        {
            int nw=q[i];
            sm0[nw]=1;sm1[nw]=rt[nw];
            for(int j=1;j<=26;j++) if(t[nw].son[j])
            {
                sm0[nw]+=sm0[t[nw].son[j]];
                sm1[nw]+=sm1[t[nw].son[j]];
            }
        }
    }
    void ffind(int opt,int k)
    {
        int sm,nw=1;
        k++;
        while(1)
        {
            sm=opt?rt[nw]:1;
            for(int i=1;i<=26;i++) if(t[nw].son[i])
            {
                int ss=sm;
                if(!opt) sm+=sm0[t[nw].son[i]];
                else sm+=sm1[t[nw].son[i]];
                if(sm>=k) {k-=ss;printf("%c",'a'+i-1);nw=t[nw].son[i];break;}
            }
            if(!opt&&k==1) break;
            if(opt&&k<=rt[nw]) break;
        }
        printf("
");
    }
}sam;

char s[Maxn];

int main()
{
    scanf("%s",s);
    int l=strlen(s);
    sam.last=sam.tot=1;
    for(int i=0;i<l;i++) sam.extend(s[i]-'a'+1);
    sam.init();
    int opt,k;
    scanf("%d%d",&opt,&k);
    if(!opt&&k+1>sm0[1]) printf("-1
");
    else if(opt&&k+rt[1]>sm1[1]) printf("-1
");
    else sam.ffind(opt,k);
    return 0;
}

2017-04-17 13:59:36