您的位置:首页 > 其它

BZOJ 4310: 跳蚤

2016-06-17 18:41 281 查看
(我当然不是因为标题才点进这道题的啦(打死不承认)) 

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

于是我们考虑一共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
,b
,c
,*x,*y,sa
,h
,rk
,n,k;
char s
;
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
[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做法?
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: