BZOJ_1717_[Usaco2006 Dec]Milk Patterns 產奶的模式_後綴數組
阿新 • • 發佈：2018-04-30
++ 模式 AI def 存在 clas ret build ray
1
2
3
2
3
2
3
1
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 21
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);
}
BZOJ_1717_[Usaco2006 Dec]Milk Patterns 產奶的模式_後綴數組