[TJOI2015]弦论

#include <bits/stdc++.h>
using namespace std;
#define int long long
typedef long long ll;
const int N=1000010;
int tot=1, last=1;
struct Node{
    int len, fa;
    int ch[26];
}node[N];
char str[N];
ll f[N], ans;ll c[N],q[N],sz[N],sm[N];
void extend(int c){
    int p=last, np=last=++tot;
    f[tot]=1;sz[tot]=1;
    node[np].len=node[p].len+1;
    for(; p && !node[p].ch[c]; p=node[p].fa)
        node[p].ch[c]=np;
    if(!p) node[np].fa=1;
    else {
        int q=node[p].ch[c];
        if(node[q].len==node[p].len+1) 
            node[np].fa=q;
        else {
            int nq=++tot;
            node[nq]=node[q], node[nq].len=node[p].len+1;
            node[q].fa=node[np].fa=nq;
            for( ; p && node[p].ch[c]==q; p=node[p].fa)
                node[p].ch[c]=nq;
        }
    }

}
void run(int t){
	for(int i=1;i<=tot;i++) c[node[i].len]++; 
	for(int i=1;i<=tot;i++) c[i]+=c[i-1];
	for(int i=1;i<=tot;i++) q[c[node[i].len]--]=i;
	for(int i=tot;i>=1;i--) sz[node[q[i]].fa]+=sz[q[i]]; //①
	for(int i=1;i<=tot;i++) sm[i]=t?sz[i]:(sz[i]=1); //②
	sz[1]=sm[1]=0;
	for(int i=tot;i>=1;i--)
		for(int c=0;c<26;c++) sm[q[i]]+=sm[node[q[i]].ch[c]]; //③
}
void Print(int p,int k){
	if(k<=sz[p]) return;
	k-=sz[p];
	for(int c=0;c<26;c++)if(node[p].ch[c]){
		if(k>sm[node[p].ch[c]]) k-=sm[node[p].ch[c]];
		else return void((putchar(c+'a'),Print(node[p].ch[c],k)));
	}
}
int t,k;
int32_t main(){
    scanf("%s", str);
    for(int i=0; str[i]; i++) extend(str[i]-'a');
    scanf("%lld%lld",&t,&k);
	run(t);
	if(sm[1]<k){printf("-1\n");}
	else {Print(1,k);}
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值