【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[],step;
 }t[Maxn*];
 int rt[Maxn*],sm0[*Maxn],sm1[*Maxn];
 int v[*Maxn],q[*Maxn];

 struct sam
 {
     int last,tot;
     void extend(int k)
     {
         int np=++tot,p=last;
         t[np].step=t[last].step+;
         rt[np]=;
         while(p&&!t[p].son[k])
         {
             t[p].son[k]=np;
             p=t[p].pre;
         }
         if(!p) t[np].pre=;
         else
         {
             int q=t[p].son[k];
             if(t[q].step==t[p].step+) 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+;
                 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=;i<=tot;i++) v[t[i].step]++;
         for(int i=;i<=tot;i++) v[i]+=v[i-];
         for(int i=tot;i>=;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>=;i--)
         {
             int nw=q[i];
             rt[t[nw].pre]+=rt[nw];
         }rt[]=;

         for(int i=tot;i>=;i--)
         {
             int nw=q[i];
             sm0[nw]=;sm1[nw]=rt[nw];
             for(int j=;j<=;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=;
         k++;
         while()
         {
             sm=opt?rt[nw]:;
             for(int i=;i<=;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-);nw=t[nw].son[i];break;}
             }
             if(!opt&&k==) break;
             if(opt&&k<=rt[nw]) break;
         }
         printf("\n");
     }
 }sam;

 char s[Maxn];

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

2017-04-17 13:59:36