BZOJ_1717_[Usaco2006 Dec]Milk Patterns 产奶的模式_后缀数组
Description
农夫John发现他的奶牛产奶的质量一直在变动。经过细致的调查,他发现:虽然他不能预见明天产奶的质量,但连续的若干天的质量有很多重叠。我们称之为一个“模式”。 John的牛奶按质量可以被赋予一个0到1000000之间的数。并且John记录了N(1<=N<=20000)天的牛奶质量值。他想知道最长的出现了至少K(2<=K<=N)次的模式的长度。比如1 2 3 2 3 2 3 1 中 2 3 2 3出现了两次。当K=2时,这个长度为4。
Input
* Line 1: 两个整数 N,K。
* Lines 2..N+1: 每行一个整数表示当天的质量值。
Output
* Line 1: 一个整数:N天中最长的出现了至少K次的模式的长度
Sample Input
8 2
1
2
3
2
3
2
3
1
1
2
3
2
3
2
3
1
Sample Output
4
先把权值离散化。
二分答案x,转化为判断是否存在一个长度为x的子串在整个字符串中出现了k次,k次可重叠。
把后缀分组,保证任意一个组里任意一个height值都不小于x。
这个长度为x的子串只可能出现在这每个组里。
直接判断是否有一组后缀个数大于等于k即可。
代码:
#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
#define N 20050
struct A {
int num,id,v;
}a[N];
bool cmp1(const A &x,const A &y){return x.num<y.num;}
bool cmp2(const A &x,const A &y){return x.id<y.id;}
int wa[N],wb[N],ws[N],wv[N],sa[N],r[N],n,m,k;
int rank[N],height[N];
void build_suffix_array() {
int i,j,p,*x=wa,*y=wb,*t;
for(i=0;i<m;i++) ws[i]=0;
for(i=0;i<n;i++) ws[x[i]=r[i]]++;
for(i=1;i<m;i++) ws[i]+=ws[i-1];
for(i=n-1;i>=0;i--) sa[--ws[x[i]]]=i;
for(j=p=1;p<n;j<<=1,m=p) {
for(p=0,i=n-j;i<n;i++) y[p++]=i;
for(i=0;i<n;i++) if(sa[i]-j>=0) y[p++]=sa[i]-j;
for(i=0;i<n;i++) wv[i]=x[y[i]];
for(i=0;i<m;i++) ws[i]=0;
for(i=0;i<n;i++) ws[wv[i]]++;
for(i=1;i<m;i++) ws[i]+=ws[i-1];
for(i=n-1;i>=0;i--) sa[--ws[wv[i]]]=y[i];
for(t=x,x=y,y=t,x[sa[0]]=0,i=p=1;i<n;i++) {
if(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+j]==y[sa[i-1]+j]) x[sa[i]]=p-1;
else x[sa[i]]=p++;
}
}
for(i=1;i<n;i++) rank[sa[i]]=i;
for(i=p=0;i<n-1;height[rank[i++]]=p) {
for(p?p--:0,j=sa[rank[i]-1];r[i+p]==r[j+p];p++);
}
}
bool check(int x) {
if(!x) return 1;
/*int lst=0,i;
for(i=1;i<=n;i++) {
if(height[i]<x) {
if(i-lst>=k) return 1;
lst=i;
}
} */
int cnt=0,i;
for(i=1;i<=n;i++) {
if(height[i]<x) {
if(cnt>=k) return 1;
cnt=1;
}else cnt++;
}
if(cnt>=k) return 1;
return 0;
}
int main() {
scanf("%d%d",&n,&k);
int i,j;
for(i=0;i<n;i++) scanf("%d",&a[i].num),a[i].id=i;
sort(a,a+n,cmp1);
for(i=0,j=0;i<n;i++) {
if(i==0||a[i].num!=a[i-1].num) j++; a[i].v=j;
}
m=j;
sort(a,a+n,cmp2);
for(i=0;i<n;i++) r[i]=a[i].v;
n++;m++;
build_suffix_array();
int l=1,r=n+1;
while(l<r) {
int mid=(l+r)>>1;
if(check(mid)) l=mid+1;
else r=mid;
}
printf("%d\n",l-1);
}