3998: [TJOI2015]弦论

题解：

每个点的size值就是这个从根-它出现的次数

如果相同只算一次就全部赋值为1就可以了

代码：

#include <bits/stdc++.h>
#define ll long long
#define rint register ll
#define rep(i,h,t) for (rint i=h;i<=t;i++)
#define dep(i,t,h) for (rint i=t;i>=h;i--)
using namespace std;
const ll N=1.1e6;
char s[N];
ll size[N],len[N],ch[N][],sum[N];
ll lst=,node=,t[N],a[N],fa[N],T,k;
void extend(ll c)
{
  ll f=lst,p=++node; lst=p;
  len[p]=len[f]+; size[p]=;
  while (f&&!ch[f][c]) ch[f][c]=p,f=fa[f];
  if (!f) { fa[p]=; return;};
  ll x=ch[f][c],y=++node;
  if (len[f]+==len[x]) {fa[p]=x; node--;return;};
  len[y]=len[f]+; fa[y]=fa[x]; fa[x]=fa[p]=y;
  memcpy(ch[y],ch[x],sizeof(ch[x]));
  while (f&&ch[f][c]==x) ch[f][c]=y,f=fa[f];
}
void dfs(ll x)
{
  if (k<=size[x])
  {
    k=;
    return;
  }
  k-=size[x];
  rep(i,,)
  {
    ll y=ch[x][i];
    if (k<=sum[y])
    {
      char cc=i+'a';
      cout<<cc;
      dfs(y);
      return;
    } else k-=sum[y];
  }
}
int main()
{
  freopen("1.in","r",stdin);
  freopen("1.out","w",stdout);
  ios::sync_with_stdio(false);
  cin>>s;
  cin>>T>>k;
  ll l=strlen(s);
  rep(i,,l) extend(s[i-]-'a');
  rep(i,,node) t[len[i]]++;
  rep(i,,node) t[i]+=t[i-];
  rep(i,,node) a[t[len[i]]--]=i;
  dep(i,node,)
  {
    ll x=a[i];
    if (T) size[fa[x]]+=size[x]; else size[x]=;
  }
  size[]=size[]=;
  dep(i,node,)
  {
    ll x=a[i];
    sum[x]+=size[x];
    rep(j,,) sum[x]+=sum[ch[x][j]];
  }
  dfs();
  if (k) cout<<-<<endl;
  return ;
}