正題

題目鏈接:https://www.luogu.com.cn/problem/P3975

---

題目大意

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

當$T=0$時，相同的子串算一個

當$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; }

總結

以上是生活随笔為你收集整理的P3975-[TJOI2015]弦论【SAM】的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。