[bzoj3998]弦论
建后缀自动机之后初始化每个点和其能到达的点表示的最小和最大的rank后在SAM上跑一遍即可
T=0/1的时候要分类讨论
- 代码
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e6+5,SZ=26;
int T;
ll k;
struct SuffixAutomaton
{
struct NODE{
int ch[SZ],fa;
int len,right;
}t[N];
int root,last,cnt;
inline int newnode(int _len=0){
t[++cnt].len=_len;
return cnt;
}
inline void init(){//"初始化
root=last=newnode(0);
}
inline void ins(int c)//"c=(char)-'a';
{
int nq=newnode(t[last].len+1),q=last;
for(;q&&!t[q].ch[c];q=t[q].fa)t[q].ch[c]=nq;
if(q==0) t[nq].fa=root;
else if(t[t[q].ch[c]].len==t[q].len+1)t[nq].fa=t[q].ch[c];
else{
int np=newnode(t[q].len+1),p=t[q].ch[c];
memcpy(t[np].ch,t[p].ch,sizeof(t[p].ch));
t[np].fa=t[p].fa;
t[p].fa=t[nq].fa=np;
for(;q&&t[q].ch[c]==p;q=t[q].fa)t[q].ch[c]=np;
}
t[nq].right=1,last=nq;
}
int topo[N],buc[N];
inline void toposet(){
int mx=0;
for(int i=1;i<=cnt;i++)buc[t[i].len]++,mx=max(mx,t[i].len);
for(int i=1;i<=mx;i++)buc[i]+=buc[i-1];
for(int i=1;i<=cnt;i++)topo[buc[t[i].len]--]=i;
}
inline void cal_right()
{
toposet();
for(int i=cnt;i;i--){
int cur=topo[i];
t[t[cur].fa].right+=t[cur].right;
}
}
ll aft[N];
inline void solve(){
t[1].right=1;
for(int i=cnt;i;i--){
int cur=topo[i];
aft[cur]=T==0?1:1ll*t[cur].right;
for(int j=0;j<SZ;j++)if(t[cur].ch[j]){
if(T==0)aft[cur]+=1ll*aft[t[cur].ch[j]];
else aft[cur]=aft[cur]+1ll*aft[t[cur].ch[j]];
if(aft[cur]>k||aft[cur]<0)aft[cur]=k+1;
}
}
int cur=1;
ll nowk=0;
while(true){
ll ad;
if(T==0)ad=1;
else ad=t[cur].right;
if(nowk+ad>k) return;
for(int i=0;i<SZ;i++)if(t[cur].ch[i]){
ll newnow=nowk+ad;
if(newnow<=k&&newnow+aft[t[cur].ch[i]]>k){
printf("%c",'a'+i);
cur=t[cur].ch[i],nowk=newnow;
break;
}
ad+=aft[t[cur].ch[i]];
}
}
}
}SAM;
char str[N];
int main()
{
SAM.init();
scanf("%s",str+1);
int len=strlen(str+1);
for(int i=1;i<=len;i++)SAM.ins(str[i]-'a');
SAM.cal_right();
scanf("%d%lld",&T,&k);
SAM.solve();
return 0;
}