【bzoj3998】[TJOI2015]弦论

建出后缀自动机后统计从每个点开始有多少子串 相当于在后缀自动机上统计路径条数
对于T=0时 right集合的大小为1
对于T=1时 right集合的大小需要统计 拓扑排序就行了
看黄学长博客学到了一个机智的拓扑排序的方法
在后缀自动机上len更大的点拓扑序一定在后面 编号更大的点拓扑序可能在后面 所以就可以这么写啦~

 for(int i=1;i<=cnt;i++)v[len[i]]++;
 for(int i=1;i<=n;i++)v[i]+=v[i-1];//长度小于等于i的串的有多少个
 也就是长度为i的一些点应在的位置起始位置
 for(int i=cnt;i>=1;i--)q[v[len[i]]--]=i;

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<map>
#include<queue>
using namespace std;
const int N=1000011;
char s[N];
int head[N],next[N],key[N],tot; 
int last=1,cnt=1;
int T,k,n;

void add(int x,int y)
{
    tot++;
    next[tot]=head[x];
    head[x]=tot;
    key[tot]=y;
}

struct node
{
    int tr[N][26],fa[N],len[N],val[N];
    void insert(int x)
    {
        int p=last;int np;np=last=++cnt;len[np]=len[p]+1;val[np]=1;
        while(p!=0&&tr[p][x]==0)tr[p][x]=np,p=fa[p];
        if(p==0)fa[np]=1;
        else
        {
           int q=tr[p][x];
           if(len[q]==len[p]+1)fa[np]=q;//
           else
           {
             int nq=++cnt;len[nq]=len[p]+1;
             memcpy(tr[nq],tr[q],sizeof(tr[q]));
             fa[nq]=fa[q],fa[np]=fa[q]=nq;
             while(tr[p][x]==q)tr[p][x]=nq,p=fa[p];//
           }
        }
    }
    int v[N],q[N],sum[N];
    void toposort()
    { 
        for(int i=1;i<=cnt;i++)v[len[i]]++;
        for(int i=1;i<=n;i++)v[i]+=v[i-1]; 
        for(int i=cnt;i>=1;i--)q[v[len[i]]--]=i;
        for(int i=cnt;i>=1;i--)
        {
          int t=q[i];
          if(T==1)val[fa[t]]+=val[t];
          else val[t]=1;    
        }
        val[1]=0;
        for(int i=cnt;i>=1;i--)
        {
            int t=q[i];sum[t]=val[t];
            for(int j=0;j<26;j++)sum[t]+=sum[tr[t][j]];
        }
    }
    void dfs(int x,int k)
    {
       if(k<=val[x])return;
       else
       {
         k-=val[x];
         for(int i=0;i<26;i++)
         if(tr[x][i])
         {
            int y=tr[x][i];
            if(k<=sum[y])
            {
                putchar(i+'a');
                dfs(y,k);
                return;
            }
            k-=sum[y];
         }

       }
    }
}sam;
int main()
{
   scanf("%s",s+1);n=strlen(s+1);
   for(int i=1;i<=n;i++)sam.insert(s[i]-'a');
   cin>>T>>k;
   sam.toposort();
   sam.dfs(1,k);
}