[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[],A[],B[],U[],Tmp[],SA[],H[];
 map <int,int>    Map;

 void    Calc_H(const int n,const int * Rank)
 {
     int    i,j,k=;
     for(i=;i<n;H[Rank[i++]]=k)
         for(k?k--:,j=SA[Rank[i]-];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=;i<n;++i)U[i]=;
     for(i=;i<n;++i)U[x[i]=str[i]]++;
     for(i=;i<m;++i)U[i]+=U[i-];
     for(i=n-;i>=;--i)SA[--U[x[i]]]=i;

     for(j=,p=;p<n;j<<=,m=p)
     {
         for(p=,i=n-j;i<n;++i)y[p++]=i;
         for(i=;i<n;++i)if(SA[i]>=j)y[p++]=SA[i]-j;
         for(i=;i<n;++i)Tmp[i]=x[y[i]];
         for(i=;i<m;++i)U[i]=;
         for(i=;i<n;++i)U[Tmp[i]]++;
         for(i=;i<m;++i)U[i]+=U[i-];
         for(i=n-;i>=;--i)SA[--U[Tmp[i]]]=y[i];
         for(swap(x,y),p=,x[SA[]]=,i=;i<n;++i)
         x[SA[i]]=cmp(y,SA[i-],SA[i],j)?p-:p++;
     }

     Calc_H(n,x);
     return x;
 }

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

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

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

     Get_SA(n+,);

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

     printf("%d\n",l);

     return ;
 }