先求一下SA
本质不同的子串个数是\( \sum n-sa[i]+1-he[i] \),按字典序二分子串,判断的时候贪心,也就是从后往前扫字符串,如果当前子串串字典序大于二分的mid子串就切一下,然后计一共有多少段
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=100005;
int n,k,b[N],wa[N],wb[N],wv[N],wsu[N],sa[N],rk[N],he[N],st[20][N<<1],ls,rs;
char s[N];
bool cmp(int r[],int a,int b,int l)
{
return r[a]==r[b]&&r[a+l]==r[b+l];
}
void saa(char r[],int n,int m)
{
int *x=wa,*y=wb;
for(int i=0;i<=m;i++)
wsu[i]=0;
for(int i=1;i<=n;i++)
wsu[x[i]=r[i]]++;
for(int i=1;i<=m;i++)
wsu[i]+=wsu[i-1];
for(int i=n;i>=1;i--)
sa[wsu[x[i]]--]=i;
for(int j=1,p=1;j<=n&&p<n;j<<=1,m=p)
{
p=0;
for(int i=n-j+1;i<=n;i++)
y[++p]=i;
for(int i=1;i<=n;i++)
if(sa[i]>j)
y[++p]=sa[i]-j;
for(int i=1;i<=n;i++)
wv[i]=x[y[i]];
for(int i=0;i<=m;i++)
wsu[i]=0;
for(int i=1;i<=n;i++)
wsu[wv[i]]++;
for(int i=1;i<=m;i++)
wsu[i]+=wsu[i-1];
for(int i=n;i>=1;i--)
sa[wsu[wv[i]]--]=y[i];
swap(x,y);
x[sa[1]]=1;
p=1;
for(int i=2;i<=n;i++)
x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p:++p;
}
for(int i=1;i<=n;i++)
rk[sa[i]]=i;
for(int i=1,j,k=0;i<=n;he[rk[i++]]=k)
for(k?k--:0,j=sa[rk[i]-1];r[i+k]==r[j+k];k++);
for(int i=1;i<=n;i++)
st[0][i]=he[i];
for(int i=1;(1<<i)<=n;i++)
for(int j=1;j+(1<<i)-1<=n;j++)
st[i][j]=min(st[i-1][j],st[i-1][j+(1<<(i-1))]);
}
int ques(int x,int y)
{
if(x==y)
return n-x+1;
int l=min(rk[x],rk[y])+1,r=max(rk[x],rk[y]),k=b[r-l+1];
if(l>r||r>n)
return 0;
return min(st[k][l],st[k][r-(1<<k)+1]);
}
void gtk(long long k)
{
for(int i=1;i<=n;i++)
{
long long nw=n-sa[i]+1-he[i];
if(nw<k)
k-=nw;
else
{
ls=sa[i],rs=ls+he[i]+k-1;
break;
}
}
}
bool bj(int l1,int r1,int l2,int r2)
{
int len1=r1-l1+1,len2=r2-l2+1,len=ques(l1,l2);
if(len>=len2&&len1>len2)
return 1;
if(len>=len1&&len2>=len1)
return 0;
if(len>=len1&&len>=len2)
return len1>len2;
return s[l1+len]>s[l2+len];
}
bool ok(long long w)
{
gtk(w);
int con=1,la=n;
for(int i=n;i>=1;i--)
{
if(s[i]>s[ls])
return 0;
if(bj(i,la,ls,rs))
con++,la=i;
if(con>k)
return 0;
}
return 1;
}
int main()
{
scanf("%d%s",&k,s+1);
n=strlen(s+1);
b[0]=-1;
for(int i=1;i<=n;i++)
b[i]=b[i>>1]+1;
saa(s,n,200);
long long l=1,r=0,ans=0;
for(int i=1;i<=n;i++)
r+=n-sa[i]+1-he[i];//cerr<<r<<endl;
ans=r;
// for(int i=1;i<=n;i++)
// cerr<<sa[i]<<" "<<he[i]<<endl;
while(l<=r)
{
long long mid=(l+r)>>1;
if(ok(mid))
r=mid-1,ans=mid;
else
l=mid+1;
}
gtk(ans);
for(int i=ls;i<=rs;i++)
printf("%c",s[i]);
return 0;
}