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]:-;
    int b2=b+k<n?y[b+k]:-;
    return a1==b1&&a2==b2;
}
void DA(){
    int *x=t1,*y=t2,m=;n++;
    sort(r,r+n);
    for(int i=;i<n;i++) sa[i]=r[i].id;
    x[sa[]]=;
    for(int i=;i<n;i++) x[sa[i]]=(r[i].x==r[i-].x?m-:m++);
    for(int k=,p=;k<=n;k*=,m=p,p=){
        for(int i=n-k;i<n;i++) y[p++]=i;
        for(int i=;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;
        for(int i=;i<m;i++) c[i]=;
        for(int i=;i<n;i++) c[x[y[i]]]++;
        for(int i=;i<m;i++) c[i]+=c[i-];
        for(int i=n-;~i;i--) sa[--c[x[y[i]]]]=y[i];

        swap(x,y);p=;x[sa[]]=;
        for(int i=;i<n;i++)
            if(cmp(y,sa[i-],sa[i],k)) x[sa[i]]=p-;
            else x[sa[i]]=p++;
        if(p>=n) break;
    }
    for(int i=;i<n;i++) rk[sa[i]]=i;n--;
}
void get_ht(){
    for(int i=,k=;i<n;ht[rk[i++]]=k){
        int j=sa[rk[i]-];k=k?k-:;
        while(s[i+k]==s[j+k]) k++;
    }
}
bool check(int mid){
    int cnt=;
    for(int i=;i<=n;i++){
        if(ht[i]<mid) cnt=;
        else cnt++;
        if(cnt>=K) return true;
    }
    return false;
}
int main(){
    while(scanf("%d%d",&n,&K)!=EOF){
        for(int i=;i<n;i++) {
            scanf("%d",&s[i]);
            r[i].x=s[i]+;r[i].id=i;
        }
        r[n].x=,r[n].id=n;
        DA();get_ht();
        int l=,r=n,ans;
        while(l<=r){
            int mid=l+r>>;
            if(check(mid)) ans=mid,l=mid+;
            else r=mid-;
        }
        printf("%d\n",ans);
    }
    return ;
}