正题

题目链接:https://www.luogu.com.cn/problem/P3975

---

题目大意

给一个字符串$s$和$t,k$。求字符串$s$第$k$大的子串。

1. 当$T=0$时，相同的子串算一个
2. 当$T=1$时，不同位置的相同子串算不同的串

---

解题思路

当$T=0$时很简单，对于每个位置求该位置开始能到多少个串即可，也就是$$f_x=\sum _{x->y}(f_y+1)$$然后根据$f_x$像权值线段树那样走就好了

当$T=1$时，我们需要求出每个$endpos$的大小，因为对于走到该节点的一个子串，显然在该位置的$endpos$的集合为结尾也有一个一样的子串，因为每个$endpos$包含所有的子节点的$endpos$，所以我们对于一个原串的后缀让$siz+1$，之后让每个节点加上它$parents$树上节点的$siz$就好。

用桶排就可以做到$O(n)$

---

$code$

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=1e6+1;
int n,las,cnt,k,t,c[N],fa[N],sum[N];
int siz[N],ch[N][26],a[N],len[N];
char s[N];
void add(int c){int p=las;int np=las=++cnt;len[np]=len[p]+1;siz[cnt]++;for(;p&&!ch[p][c];p=fa[p])ch[p][c]=np;if(!p)fa[np]=1;else{int q=ch[p][c];if(len[p]+1==len[q])fa[np]=q;else{int nq=++cnt;len[nq]=len[p]+1;memcpy(ch[nq],ch[q],sizeof(ch[nq]));fa[nq]=fa[q];fa[np]=fa[q]=nq;for(;p&&ch[p][c]==q;p=fa[p])ch[p][c]=nq;}}return;
}
void solve(int x,int k){if(k<=siz[x])return;k-=siz[x];for(int i=0;i<26;i++){if(!ch[x][i])continue;if(k>sum[ch[x][i]])k-=sum[ch[x][i]];else {printf("%c",i+'a');solve(ch[x][i],k);return;}}return;
}
int main()
{cnt=las=1;scanf("%s",s+1);n=strlen(s+1);for(int i=1;i<=n;i++)add(s[i]-'a');scanf("%d%d",&t,&k);for(int i=1;i<=cnt;i++)c[len[i]]++;for(int i=1;i<=n;i++)c[i]+=c[i-1];for(int i=1;i<=cnt;i++)a[c[len[i]]--]=i;for(int i=cnt;i>=1;i--)siz[fa[a[i]]]+=siz[a[i]];for(int i=1;i<=cnt;i++)t?(sum[i]=siz[i]):(sum[i]=siz[i]=1);siz[1]=siz[0]=sum[1]=sum[0]=0;for(int i=cnt;i>=1;i--)for(int j=0;j<26;j++)if(ch[a[i]][j])sum[a[i]]+=sum[ch[a[i]][j]];	if(sum[1]<k)printf("-1");else solve(1,k);return 0;
}