bzoj 3998 弦论 —— 后缀自动机

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=3998

关于相同子串算一个还是算多个，其实就是看一种状态的 right 集合是否加上 Parent 树中子树的 right 集合；

虽然是子串却不管 l 数组，因为实际上 dfs 走到一个点，得到的是一个确定的子串，而这个子串的状态属于这个点表示的状态，l 数组是这个点表示的状态数，当然不用考虑；

记一个 sum 表示这个状态往后加字母能得到的所有子串个数，然后在SAM上按字典序 dfs ，走过的边就是添加的字符。

代码如下：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int const xn=5e5+;
int T,K,fa[xn<<],l[xn<<],go[xn<<][],lst=,cnt=,siz[xn<<],sum[xn<<];
int tax[xn<<],t[xn<<];
char s[xn];
void add(int w)
{
  int p=lst,np=++cnt; lst=np; l[np]=l[p]+; siz[np]=;
  for(;p&&!go[p][w];p=fa[p])go[p][w]=np;
  if(!p)fa[np]=;
  else
    {
      int q=go[p][w];
      if(l[q]==l[p]+)fa[np]=q;
      else
    {
      int nq=++cnt; l[nq]=l[p]+;
      memcpy(go[nq],go[q],sizeof go[q]);
      fa[nq]=fa[q]; fa[q]=fa[np]=nq;
      for(;go[p][w]==q;p=fa[p])go[p][w]=nq;
    }
    }
}
void Tsort()
{
  for(int i=;i<=cnt;i++)tax[l[i]]++;
  for(int i=;i<=cnt;i++)tax[i]+=tax[i-];
  for(int i=;i<=cnt;i++)t[tax[l[i]]--]=i;
  for(int i=cnt;i;i--)
    {
      int x=t[i];
      if(T==)siz[x]=;
      else siz[fa[x]]+=siz[x];
    }
  siz[]=;//
  for(int i=cnt;i;i--)
    {
      int x=t[i]; sum[x]=siz[x];
      for(int j=;j<;j++)
    if(go[x][j])sum[x]+=sum[go[x][j]];
    }
}
void dfs(int x,int s)
{
  if(s<=siz[x])return;//<=
  s-=siz[x];
  for(int j=;j<;j++)
    {
      if(!go[x][j])continue;
      if(s>sum[go[x][j]])s-=sum[go[x][j]];
      else
    {
      putchar(j+'a');
      dfs(go[x][j],s);
      return;
    }
    }
}
int main()
{
  scanf("%s",s+); int n=strlen(s+);
  scanf("%d%d",&T,&K);
  for(int i=;i<=n;i++)add(s[i]-'a');
  Tsort(); dfs(,K); puts("");
  return ;
}