BZOJ_1717_[Usaco2006 Dec]Milk Patterns 产奶的模式_后缀数组

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

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);
}