POJ3261 Milk Patterns（二分+后缀数组）

题目求最长的重复k次可重叠子串。

与POJ1743同理。

1. 二分枚举ans判定是否成立
2. height分组，如果大于等于ans的组里的个数大于等于k-1，这个ans就可行

 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 using namespace std;
 #define MAXN 1000001

 int wa[MAXN],wb[MAXN],wv[MAXN],ws[MAXN];
 int cmp(int *r,int a,int b,int l){
     return r[a]==r[b] && r[a+l]==r[b+l];
 }
 int sa[MAXN],rank[MAXN],height[MAXN];
 void SA(int *r,int n,int m){
     int *x=wa,*y=wb;

     for(int i=; i<m; ++i) ws[i]=;
     for(int i=; i<n; ++i) ++ws[x[i]=r[i]];
     for(int i=; i<m; ++i) ws[i]+=ws[i-];
     for(int i=n-; i>=; --i) sa[--ws[x[i]]]=i;

     int p=;
     for(int j=; p<n; j<<=,m=p){
         p=;
         for(int i=n-j; i<n; ++i) y[p++]=i;
         for(int i=; i<n; ++i) if(sa[i]>=j) y[p++]=sa[i]-j;
         for(int i=; i<n; ++i) wv[i]=x[y[i]];
         for(int i=; i<m; ++i) ws[i]=;
         for(int i=; i<n; ++i) ++ws[wv[i]];
         for(int i=; i<m; ++i) ws[i]+=ws[i-];
         for(int i=n-; i>=; --i) sa[--ws[wv[i]]]=y[i];
         swap(x,y); x[sa[]]=; p=;
         for(int i=; i<n; ++i) x[sa[i]]=cmp(y,sa[i-],sa[i],j)?p-:p++;
     }

     for(int i=; i<n; ++i) rank[sa[i]]=i;
     int k=;
     for(int i=; i<n-; height[rank[i++]]=k){
         if(k) --k;
         for(int j=sa[rank[i]-]; r[i+k]==r[j+k]; ++k);
     }
 }

 int n,k,a[MAXN];
 bool isok(int len){
     int cnt=;
     bool flag=;
     for(int i=; i<=n; ++i){
         if(height[i]>=len){
             if(flag){
                 ++cnt;
                 if(cnt+>=k) return ;
             }else{
                 flag=;
                 cnt=;
                 if(cnt+>=k) return ;
             }
         }else{
             flag=;
             cnt=;
         }
     }
     return ;
 }
 int main(){
     int mx=;
     scanf("%d%d",&n,&k);
     for(int i=; i<n; ++i){
         scanf("%d",a+i);
         mx=max(mx,++a[i]);
     }
     a[n]=;
     SA(a,n+,mx+);
     int l=,r=n;
     while(l<r){
         int mid=l+r+>>;
         if(isok(mid)) l=mid;
         else r=mid-;
     }
     printf("%d",l);
     return ;
 }