bzoj1717: [Usaco2006 Dec]Milk Patterns 产奶的模式(后缀数组+二分)

/*
求可重叠的至少重复K次的最长字串
以1为下标起点，因为a[i]最大到1000000，所以要先离散一下
二分长度len
然后O(n)检验
后看h[i]是否有连续的一段h[i]大于len的，并且h[i]连续的长度大于K则满足 
*/
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int maxn = 20010;
int c[maxn],sa[maxn],t1[maxn],t2[maxn],n,m,K,tot,t[maxn],num[maxn],a[maxn],rank[maxn],height[maxn];

bool cmp(int *r, int a, int b, int l){
    return r[a]==r[b] && r[a+l]==r[b+l];
}

void suffix(int *a, int n, int m){
    int p=0, *x=t1, *y=t2;
    for (int i=0; i<=m; i++) c[i]=0;
    for (int i=1; i<=n; i++) c[x[i]=a[i]]++;
    for (int i=1; i<=m; i++) c[i]+=c[i-1];
    for (int i=n; i>=1; i--) sa[c[x[i]]--]=i;
    for (int j=1; j<=n; j<<=1){
        p=0;
        for (int i=n-j+1; i<=n; i++) y[++p]=i;
        for (int i=1; i<=n; i++) if (sa[i]>j) y[++p]=sa[i]-j;
        for (int i=0; i<=m; i++) c[i]=0;
        for (int i=1; i<=n; i++) c[x[y[i]]]++;
        for (int i=1; i<=m; i++) c[i]+=c[i-1];
        for (int i=n; i>=1; i--) sa[c[x[y[i]]]--]=y[i];
        swap(x,y); x[sa[1]]=1; p=1;
        for (int i=2; i<=n; i++)
            x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p:++p;
        m=p; if (p>=n) break;
    }
    //for (int i=1; i<=n; i++) printf("%d
", sa[i]);
}

void get_height(int *a){
    int k=0,j;
    for (int i=1; i<=n; i++) rank[sa[i]]=i;
    for (int i=1; i<=n; i++){
        j=sa[rank[i]-1];
        k?k--:0;
        while (a[j+k]==a[i+k]) k++;
        height[rank[i]]=k;
    }
    //for (int i=1; i<=n; i++) printf("%d
", height[i]);
}

int find(int x){
    int l=1, r=tot, ans;
    while (l<=r){
        int mid=l+r>>1;
        if (t[mid]<=x) ans=mid,l=mid+1;
        else r=mid-1;
    }
    return ans;
}

bool check(int k){
    int i,tot=0;
    for (i=1; i<=n; i++){
        if (height[i]>=k) tot++;
        else{
            if (tot+1>=K) return 1;
            else tot=0;
        }
    }
    if (tot+1>=K) return 1; return 0;
}

int main(){
    scanf("%d%d", &n, &K);
    for (int i=1; i<=n; i++) scanf("%d", &a[i]),num[i]=a[i];
    sort(num+1,num+1+n); tot=0;
    t[++tot]=num[1];
    for (int i=2; i<=n; i++)
        if (num[i]!=num[i-1])
            t[++tot]=num[i];
    for (int i=1; i<=n; i++) a[i]=find(a[i]);
    suffix(a,n,n);
    get_height(a);
    int l=1,r=n,ans;
    while (l<=r){
        int mid=l+r>>1;
        if (check(mid)) ans=mid,l=mid+1;
        else r=mid-1;
    }
    printf("%d
", ans);
    return 0;
}