Milk Patterns（poj 3261）

题意：找出出现k次的可重叠的最长子串的长度。

/*
    求出height数组，二分答案，将height分组，看有没有一组个数大于K。 
    WA了很多遍，最后看别人的题解，发现往字符串后面填了一个0，就可以AC了，不知道为什么。 
*/
#include<cstdio>
#include<iostream>
#include<algorithm>
#define N 100010
using namespace std;
int s[N],sa[N],rk[N],ht[N],t1[N],t2[N],c[N],n,K;
struct node{
    int x,id;
    bool operator<(const node&a)const{
        if(x==a.x) return id<a.id;
        return x<a.x;
    }
}r[N];
bool cmp(int *y,int a,int b,int k){
    int a1=y[a],b1=y[b];
    int a2=a+k<n?y[a+k]:-1;
    int b2=b+k<n?y[b+k]:-1;
    return a1==b1&&a2==b2;
}
void DA(){
    int *x=t1,*y=t2,m=1;n++;
    sort(r,r+n);
    for(int i=0;i<n;i++) sa[i]=r[i].id;
    x[sa[0]]=0;
    for(int i=1;i<n;i++) x[sa[i]]=(r[i].x==r[i-1].x?m-1:m++);
    for(int k=1,p=0;k<=n;k*=2,m=p,p=0){
        for(int i=n-k;i<n;i++) y[p++]=i;
        for(int i=0;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;
        for(int i=0;i<m;i++) c[i]=0;
        for(int i=0;i<n;i++) c[x[y[i]]]++;
        for(int i=1;i<m;i++) c[i]+=c[i-1];
        for(int i=n-1;~i;i--) sa[--c[x[y[i]]]]=y[i];
        
        swap(x,y);p=1;x[sa[0]]=0;
        for(int i=1;i<n;i++)
            if(cmp(y,sa[i-1],sa[i],k)) x[sa[i]]=p-1;
            else x[sa[i]]=p++;
        if(p>=n) break;
    }
    for(int i=0;i<n;i++) rk[sa[i]]=i;n--;
}
void get_ht(){
    for(int i=0,k=0;i<n;ht[rk[i++]]=k){
        int j=sa[rk[i]-1];k=k?k-1:0;
        while(s[i+k]==s[j+k]) k++;
    }
}
bool check(int mid){
    int cnt=0;
    for(int i=1;i<=n;i++){
        if(ht[i]<mid) cnt=1;
        else cnt++;
        if(cnt>=K) return true;
    }
    return false;
}
int main(){
    while(scanf("%d%d",&n,&K)!=EOF){
        for(int i=0;i<n;i++) {
            scanf("%d",&s[i]);
            r[i].x=s[i]+1;r[i].id=i;
        }
        r[n].x=0,r[n].id=n;
        DA();get_ht();
        int l=0,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;
}