bzoj3998: [TJOI2015]弦论(SAM+dfs)

3998: [TJOI2015]弦论

题目：传送门

---

题解：

　　 SAM的入门题目（很好的复习了SAM并加强Right集合的使用）

　　 其实对于第K小的字符串直接从root开始一通DFS就好，因为son边是直接根据字符存的呀，相当于自带字典序，直接从‘a' 开始找。

　　 一开始初始化一下从当前状态出发所能走的路径数（也就是能构成多少个字符串啦）

　　 然后根据T= 0/1分情况来处理Right集合大小：

　　 T== 0 ：那Right集合大小就直接全部等于1咯

　　 T== 1  ： 直接累加啊（原始操作）

　　 然后具体看代码注释吧：

　　

---

代码：

 #include<cstdio>
 #include<cstring>
 #include<cstdlib>
 #include<cmath>
 #include<algorithm>
 using namespace std;
 struct SAM
 {
     int son[],fail,dep;
 }ch[];int cnt,last,root,a[];
 void add(int k)
 {
     int x=a[k];
     int p=last,np=++cnt;ch[np].dep=k;
     while(p && ch[p].son[x]==)ch[p].son[x]=np,p=ch[p].fail;
     if(p==)ch[np].fail=root;
     else
     {
         int q=ch[p].son[x];
         if(ch[p].dep+==ch[q].dep)ch[np].fail=q;
         else
         {
             int nq=++cnt;
             ch[nq]=ch[q];ch[nq].dep=ch[p].dep+;
             ch[np].fail=ch[q].fail=nq;
             while(p && ch[p].son[x]==q)ch[p].son[x]=nq,p=ch[p].fail;
         }
     }
     last=np;
 }
 int len,T,k,Rsort[],r[],sa[],sum[];
 char s[];
 void dfs(int x)
 {
     if(k<=r[x])return ;
     k-=r[x];//要从x继续往下搜，那么结束位置为x的子串一定会更优，那就减去这样的子串个数啊，r记录的就是这个
     for(int i=;i<=;i++)
         if(ch[x].son[i]!=)
         {
             int y=ch[x].son[i];
             if(k>sum[y])k-=sum[y];//当前k如果比y可以走出来的路径还要多，那么第k小的一定不在y的路径上。由于是按照字典序小到大问的，那么y的路径一定比后面枚举的要优，那也要减，sum记录的就是可行路径数
             else {printf("%c",i+'a');dfs(y);return ;}
         }
 }
 int main()
 {
     cnt=;root=last=++cnt;
     scanf("%s",s+);len=strlen(s+);
     for(int i=;i<=len;i++)a[i]=s[i]-'a';
     for(int i=;i<=len;i++)add(i);
     scanf("%d%d",&T,&k);
     for(int i=;i<=cnt;i++)Rsort[ch[i].dep]++;
     for(int i=;i<=len;i++)Rsort[i]+=Rsort[i-];
     for(int i=cnt;i>=;i--)sa[Rsort[ch[i].dep]--]=i;
     for(int i=,p=root;i<=len;i++)p=ch[p].son[a[i]],r[p]++;
     for(int i=cnt;i>=;i--)
         if(T==)r[ch[sa[i]].fail]=;
         else r[ch[sa[i]].fail]+=r[sa[i]];
     r[root]=;
     for(int i=cnt;i>=;i--)
     {
         int now=sa[i];sum[now]=r[now];
         for(int j=;j<=;j++)if(ch[now].son[j]!=)sum[now]+=sum[ch[now].son[j]];
     }
     if(k>sum[root])printf("-1\n");
     else dfs(root);
     return ;
 }