POJ 3261 字符串上的k次覆盖问题

题目大意：

给定一个数组，求一个最大的长度的子串至少出现过k次

一个子串出现多次，也就是说必然存在2个子串间的前缀长度为所求的值

通过二分答案，通过线性扫一遍，去判断出现次数，也就是说每次遇见一个height[i] , 出现次数就加1，否则重置为1

 #include <cstdio>
 #include <cstring>
 #include <iostream>
 using namespace std;
 const int N = ;
 int rank[N] , sa[N] , height[N];
 int wa[N] , wb[N] , wsf[N] , wv[N];
 int a[N];

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

 void getSa(int *r , int *sa , int n , int m)
 {
     int i,j,p;
     int *x=wa , *y=wb , *t;
     for(i= ; i<m ; i++) wsf[i]=;
     for(i= ; i<n ; i++) wsf[x[i]=r[i]]++;
     for(i= ; i<m ; i++) wsf[i] += wsf[i-];
     for(i=n- ; i>= ; i--) sa[--wsf[x[i]]]=i;

     p=;
     for(j= ; 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++) wv[i]=x[y[i]];
         for(i= ; i<m ; i++) wsf[i]=;
         for(i= ; i<n ; i++) wsf[wv[i]]++;
         for(i= ; i<m ; i++) wsf[i]+=wsf[i-];
         for(i=n- ; i>= ; i--) sa[--wsf[wv[i]]]=y[i];

         t=x,x=y,y=t;
         x[sa[]]=;
         for(p= , i= ; i<n ; i++)
             x[sa[i]] = cmp(y , sa[i-] , sa[i] , j)?p-:p++;
     }
     return ;
 }

 void getHeight(int *r , int *sa , int n)
 {
     for(int i= ; i<=n ; i++) rank[sa[i]]=i;
     int k=;
     int j;
     for(int i= ; i<n ; height[rank[i++]]=k)
         for(k?k--: , j=sa[rank[i]-] ; r[i+k]==r[j+k] ; k++);
     return;
 }

 bool check(int m , int n , int k)
 {
     int cnt=;
     for(int i= ; i<=n ; i++){
         if(height[i]>=m){
             cnt++;
          //   cout<<"here: "<<m<<" "<<sa[i]<<" "<<sa[i-1]<<endl;
             if(cnt>=k) return true;
         }
         else {cnt=;if(cnt>=k) return true;}
     }
     return false;
 }

 int main()
 {
   //  freopen("a.in" , "r" , stdin);
     int n,k;
     while(~scanf("%d%d" , &n , &k))
     {
         int maxn = ;
         for(int i= ; i<n ; i++){
             scanf("%d" , a+i);
             a[i]++;
             maxn = max(maxn , a[i]);
         }
         a[n]=;
         getSa(a , sa , n+ , maxn+);
         getHeight(a , sa , n);

         int l= , r=n , ans=;
         while(l <= r){
             int m=(l+r)>>;
             if(check(m,n,k)){
                 ans = m;
                 l=m+;
             }else r=m-;
         }
         printf("%d\n" , ans);
     }
     return ;
 }