poj 3261 Milk Patterns 後綴數組+二分

題意是求可以重複的出現次數超過k次的子串，最長是多少。

樣例輸入
8 2
1 2 3 2 3 2 3 1

輸出 4

二分驗證是否有連續的k-1個height大於所驗證的答案。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define SZ 20005
using namespace std;
int L[SZ],a[SZ];
int n,K;
struct Suffix
{
    int sa[SZ];
    int rank[SZ<<1],tmp[SZ];
    int height[SZ];
    int cnt[SZ],sec[SZ];
    void get_sa()
    {
        for(int i=1;i<=n;i++) rank[i]=a[i];
        for(int i=1;i<=n;i++) cnt[rank[i]]++;
        for(int i=1;i<=n;i++) cnt[i]+=cnt[i-1];
        for(int i=1;i<=n;i++) sa[cnt[rank[i]]--]=i;

        for(int k=1;k<=n;k<<=1)
        {
            int top=0;
            for(int i=n-k+1;i<=n;i++) sec[++top]=i;
            for(int i=1;i<=n;i++) if(sa[i]>k) sec[++top]=sa[i]-k;

            for(int i=0;i<=n;i++) cnt[i]=0;
            for(int i=1;i<=n;i++) cnt[rank[i]]++;
            for(int i=1;i<=n;i++) cnt[i]+=cnt[i-1];
            for(int i=n;i>=1;i--) sa[cnt[rank[sec[i]]]--]=sec[i];

            top=tmp[sa[1]]=1;
            for(int i=2;i<=n;i++)
            {
                if(rank[sa[i]]!=rank[sa[i-1]]||rank[sa[i]+k]!=rank[sa[i-1]+k]) top++;
                tmp[sa[i]]=top;
            }
            for(int i=1;i<=n;i++)
                rank[i]=tmp[i];
        }

        int k=0;
        for(int i=1;i<=n;i++)
        {
            if(k) k--;
            while(i+k<=n&&sa[rank[i]-1]+k<=n&&a[i+k]==a[sa[rank[i]-1]+k]) k++;
            height[rank[i]]=k;
        }
    }
    bool check(int mid)
    {
        int tot=0;
        for(int i=2;i<=n;i++)
        {
            if(height[i]>=mid) tot++;
            else tot=0;
            if(tot>=K-1) return true;
        }
        return false;
    }
}SA;
int main()
{
    scanf("%d%d",&n,&K);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        L[i]=a[i];
    }
    sort(L+1,L+n+1);
    int len=unique(L+1,L+n+1)-L;
    for(int i=1;i<=n;i++)
        a[i]=lower_bound(L+1,L+len+1,a[i])-L;
    SA.get_sa();
    int l=0,r=n+1;
    while(r-l>1)
    {
        int mid=(l+r)>>1;
        if(SA.check(mid)) l=mid;
        else r=mid;
    }
    printf("%d",l);
    return 0;
}