[Poj3261] [Bzoj1717] [后缀数组论文例题，USACO 2006 December Gold] Milk Patterns [后缀数组可重叠的k次最长重复子串]

和上一题(POJ1743,上一篇博客)相似，只是二分的判断条件是：是否存在一段后缀的个数不小于k

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <ctime>
#include <map>

using namespace std;

int    t;
int    str[21000],A[21000],B[21000],U[21000],Tmp[21000],SA[21000],H[21000];
map <int,int>    Map;

void    Calc_H(const int n,const int * Rank)
{
    int    i,j,k=0;
    for(i=0;i<n;H[Rank[i++]]=k)
        for(k?k--:0,j=SA[Rank[i]-1];str[i+k]==str[j+k];++k);
    return ;
}

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

int*    Get_SA(const int n,int    m)
{
    int    i,j,p,*x=A,*y=B;

    for(i=0;i<n;++i)U[i]=0;
    for(i=0;i<n;++i)U[x[i]=str[i]]++;
    for(i=1;i<m;++i)U[i]+=U[i-1];
    for(i=n-1;i>=0;--i)SA[--U[x[i]]]=i;

    for(j=1,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)y[p++]=SA[i]-j;
        for(i=0;i<n;++i)Tmp[i]=x[y[i]];
        for(i=0;i<m;++i)U[i]=0;
        for(i=0;i<n;++i)U[Tmp[i]]++;
        for(i=1;i<m;++i)U[i]+=U[i-1];
        for(i=n-1;i>=0;--i)SA[--U[Tmp[i]]]=y[i];
        for(swap(x,y),p=1,x[SA[0]]=0,i=1;i<n;++i)
        x[SA[i]]=cmp(y,SA[i-1],SA[i],j)?p-1:p++;
    }

    Calc_H(n,x);
    return x;
}

bool    Check(const int lim,const int n)
{
    int    cnt=0;
    for(int i=1;i<=n;++i)
    {
        if(i>1 && H[i]<lim)
        {
            if(cnt>=t)return true;
            cnt=1;
        }
        if(H[i]>=lim)cnt++;
    }
    if(cnt>=t)return true;
    return false;
}

int main()
{
    int    i,l,r,n,cnt=0;

    scanf("%d%d",&n,&t);
    for(i=0;i<n;++i)
    {
        scanf("%d",&str[i]);
        if(!Map[str[i]])Map[str[i]]=++cnt;
        str[i]=Map[str[i]];
    }

    Get_SA(n+1,21000);

    l=0;r=n;
    while(l<r-1)
    {
        int    mid=l+((r-l)>>1);
        if(Check(mid,n))l=mid;
        else    r=mid;
    }

    printf("%d
",l);

    return 0;
}