【bzoj3998】[TJOI2015]弦论 后缀自动机
2018-01-23 21:55
465 查看
Description
对于一个给定长度为N的字符串,求它的第K小子串是什么。
Input
第一行是一个仅由小写英文字母构成的字符串S
第二行为两个整数T和K,T为0则表示不同位置的相同子串算作一个。T=1则表示不同位置的相同子串算作多个。K的意义如题所述。
Output
输出仅一行,为一个数字串,为第K小的子串。如果子串数目不足K个,则输出-1
Sample Input
aabc
0 3
Sample Output
aab
HINT
N<=5*10^5
T<2
K<=10^9
题解
T=0 right[i]=1
T=0 right[fa[i]]+=right[i]表示这个子串出现几次
处理出每个节点之后能有多少种子串,dfs计算即可。
代码
对于一个给定长度为N的字符串,求它的第K小子串是什么。
Input
第一行是一个仅由小写英文字母构成的字符串S
第二行为两个整数T和K,T为0则表示不同位置的相同子串算作一个。T=1则表示不同位置的相同子串算作多个。K的意义如题所述。
Output
输出仅一行,为一个数字串,为第K小的子串。如果子串数目不足K个,则输出-1
Sample Input
aabc
0 3
Sample Output
aab
HINT
N<=5*10^5
T<2
K<=10^9
题解
T=0 right[i]=1
T=0 right[fa[i]]+=right[i]表示这个子串出现几次
处理出每个节点之后能有多少种子串,dfs计算即可。
代码
#include<cstdio> #include<iostream> #include<cstring> #include<algorithm> #include<queue> #define ll long long using namespace std; inline int read() { int x=0,f=1;char ch=getchar(); while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();} while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();} return x*f; } int T,n,k; char s[500005]; int last,cnt,S; int trans[1000005][26],fa[1000005],l[1000005],Right[1000005],sum[1000005]; int v[500005],q[1000005]; void add(int x) { int p=last,np=++cnt,c=s[x]-'a';last=cnt;l[np]=x;Right[np]=1; for (;p&&!trans[p][c];p=fa[p]) trans[p][c]=np; if (!p) fa[np]=S; else { int q=trans[p][c]; if (l[p]+1==l[q]) fa[np]=q; else { int nq=++cnt;l[nq]=l[p]+1; memcpy(trans[nq],trans[q],sizeof trans[q]); fa[nq]=fa[q];fa[q]=fa[np]=nq; for (;trans[p][c]==q;p=fa[p]) trans[p][c]=nq; } } } void pre() { for (int i=1;i<=cnt;i++) v[l[i]]++; for (int i=1;i<=n;i++) v[i]+=v[i-1]; for (int i=cnt;i;i--) q[v[l[i]]--]=i; for (int i=cnt;i;i--) { int t=q[i]; if (T==1) Right[fa[t]]+=Right[t]; else Right[t]=1; } Right[1]=0; for (int i=cnt;i;i--) { int t=q[i];sum[t]=Right[t]; for (int j=0;j<26;j++) sum[t]+=sum[trans[t][j]]; } } void dfs(int x,int k) { if (k<=Right[x]) return; k-=Right[x]; for (int i=0;i<26;i++)if (int t=trans[x][i]) { if (k<=sum[t]) { putchar('a'+i); dfs(t,k); return; } k-=sum[t]; } } int main() { scanf("%s",s+1);T=read();k=read(); last=S=++cnt;n=strlen(s+1); for (int i=1;i<=n;i++) add(i); pre();if (k>sum[1])puts("-1");else dfs(S,k); return 0; }
相关文章推荐
- bzoj 3998: [TJOI2015]弦论 后缀自动机
- [BZOJ3998][TJOI2015]弦论(后缀自动机)
- BZOJ 3998: [TJOI2015]弦论 后缀自动机
- bzoj 3998: [TJOI2015]弦论 后缀自动机
- BZOJ 3998 TJOI 2015 弦论 后缀自动机
- [BZOJ3998]TJOI2015弦论|后缀自动机
- 【BZOJ3998】【TJOI2015】弦论 后缀自动机
- bzoj 3998 [TJOI2015]弦论 后缀数组||后缀自动机
- [bzoj3998][TJOI2015]弦论-后缀自动机
- 【bzoj3998】[TJOI2015]弦论 后缀自动机+dp
- 【后缀自动机】 BZOJ 3998: [TJOI2015]弦论
- BZOJ 3998 TJOI2015 弦论 后缀自动机
- bzoj 3998: [TJOI2015]弦论(后缀自动机)
- 【BZOJ3998】【TJOI2015】 弦论 后缀自动机
- BZOJ_3998_[TJOI2015]弦论_后缀自动机
- [BZOJ3998][TJOI2015]弦论(后缀自动机)
- BZOJ 3998 [TJOI2015]弦论 ——后缀自动机
- BZOJ 3998 TJOI2015 弦论 后缀自动机
- [K大子串 后缀自动机 模板题] BZOJ 3998 [TJOI2015]弦论
- bzoj 3998: [TJOI2015]弦论 (后缀自动机)