BZOJ 4310: 跳蚤

(我当然不是因为标题才点进这道题的啦(打死不承认)) 

最大字典序的串最小,听起来就很像二分嘛

于是我们考虑一共m个本质不同的子串,从这里面二分就好了

首先我们需要求第k大(小?)(即二分时的mid)的子串:

我们从前往后扫sa数组,即从小到大枚举后缀,若当前后缀贡献的字串数不够k,则让k减去这些贡献,否则答案即是这些后缀所构成的贡献(就是一个后缀上的一大堆前缀区间啦)中的第k个。

然后我们考虑判断是否可行

我们从后往前扫字符串,若当前子串串字典序大于第k个子串,就割掉,最后看至少要割出几个串就好了

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<queue>
#include<cmath>
using namespace std;
#define rep(i,l,r) for(int i=l;i<=r;i++)
#define per(i,r,l) for(int i=r;i>=l;i--)
#define mmt(a,v) memset(a,v,sizeof(a))
#define tra(i,u) for(int i=head[u];i;i=e[i].next)
const int N=100000+5;
typedef long long ll;
#define equ(x) (y[sa[i]+x]==y[sa[i-1]+x])
int a[N],b[N],c[N],*x,*y,sa[N],h[N],rk[N],n,k;
char s[N];
void radix(int m){
	rep(i,1,m)c[i]=0;
	rep(i,1,n)c[x[y[i]]]++;
	rep(i,1,m)c[i]+=c[i-1];
	per(i,n,1)sa[c[x[y[i]]]--]=y[i];
}
void build(int m){
	rep(i,1,n)x[i]=s[i],y[i]=i;radix(m);
	for(int k=1,p=0;k<=n;k<<=1,m=p,p=0){
		rep(i,n-k+1,n)y[++p]=i;
		rep(i,1,n)if(sa[i]>k)y[++p]=sa[i]-k;
		radix(m);swap(x,y);x[sa[1]]=p=1;
		rep(i,2,n)x[sa[i]]=equ(0)&&equ(k)?p:++p;
		if(p==n)break;
	}
	rep(i,1,n)rk[sa[i]]=i;
	for(int i=1,k=0;i<=n;h[rk[i++]]=k)
	for(k?k--:0;i+k<=n&&s[i+k]==s[sa[rk[i]-1]+k];k++);
}
int ls,rs;
void kth(ll k){
	ll cur;
	rep(i,1,n){
		cur=n-sa[i]+1-h[i];
		if(cur<k)k-=cur;
		else{ls=sa[i];rs=ls+h[i]+k-1;break;}
	}
}
int st[N][20];
void rmq_init(){
	rep(i,1,n)st[i][0]=h[i];
	for(int j=1;(1<<j)<=n;j++)
	for(int i=1;i+(1<<j)-1<=n;i++)
	st[i][j]=min(st[i][j-1],st[i+(1<<(j-1))][j-1]);
}
int rmq(int l,int r){
	int k=log2(r-l+1);
	return min(st[l][k],st[r-(1<<k)+1][k]);
}
int lcp(int l1,int l2){
	if(l1==l2)return n-l1+1;
	int l=rk[l1],r=rk[l2];
	if(l>r)swap(l,r);
	return rmq(l+1,r);
}
bool cmp(int l1,int r1,int l2,int r2){
	int len1=r1-l1+1,len2=r2-l2+1,com=lcp(l1,l2);
	if(com>=len2&&len1>len2)return true;
	if(com>=len1&&len2>=len1)return false;
	if(com>=len1&&com>=len2)return len1>len2;
	return s[l1+com]>s[l2+com];
}
bool check(){
	int cnt=1,last=n;
	per(i,n,1){
		if(s[i]>s[ls])return false;
		if(cmp(i,last,ls,rs))cnt++,last=i;
		if(cnt>k)return false;
	}
	return true;
}
void print(){rep(i,ls,rs)putchar(s[i]);puts("");}
int main(){
	//freopen("a.in","r",stdin);
	scanf("%d",&k);
	scanf("%s",s+1);n=strlen(s+1);
	x=a;y=b;build(200);rmq_init();
	ll l=1,r=0;
	rep(i,1,n)r+=n-sa[i]+1-h[i];
	while(l<=r){
		ll mid=l+r>>1;
		kth(mid);
		if(check())r=mid-1;
		else l=mid+1;
	}
	kth(r+1);
	print();
	return 0;
}
这题似乎有SAM做法?

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值