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做法?
最大字典序的串最小,听起来就很像二分嘛
于是我们考虑一共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做法?
相关文章推荐
- 数独求解
- Latent Semantic Analysis(LSA)
- Spring MVC + Shiro 实现权限验证
- html a name属性
- Java 线程池原理和队列详解
- 如何让你的网站用discuz插件变的有力量
- 记一个mysql-connector-java包的bug
- 基于pix硬件的apm开发环境搭建
- Product Hunt红遍硅谷的秘密
- 论程序员的自我解放
- unable to rollback aborted in-doubt branch on resource
- cin,cin.get(),cin.getline()
- Java根据URL连接请求JSON数据解析代码
- js与C++交互及C++解析json
- 周五课件-文件的重定向(2)
- 性能测试调优学习链接
- Html5必看:教你如何选择移动APP开发框架
- 面向对象程序设计得分(第06回合)
- vc++>>Connection using old (pre-4.1.1) authentication protocol refused (client option 'secure_auth' enable
- 关于CreateProcess函数一些经验