BZOJ 4310: 跳蚤

来源：互联网发布：centos squid 重启编辑：程序博客网时间：2024/06/02 21:16

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

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

于是我们考虑一共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