poj3261 Milk Patterns 后缀数组求可重叠的k次最长重复子串

题目链接：http://poj.org/problem?id=3261

思路：

后缀数组的很好的一道入门题目

先利用模板求出sa数组和height数组

然后二分答案(即对于可能出现的重复长度进行二分) ，二分的时候，对 height进行分组，看是否存在一组height值使得其重复的次数大于等于k

代码如下：

 #include<iostream>
 #include<cstdlib>
 #include<cstdio>
 #include<cstring>
 using namespace std;
 #define maxn 20010
 int wa[maxn],wb[maxn],wv[maxn],wq[maxn];
 int rank[maxn],sa[maxn];
 int r[maxn];
 int height[maxn];
 int s[maxn];
 int n,k;
 int cmp(int *r,int a,int b,int l)
 {
     return r[a]==r[b]&&r[a+l]==r[b+l];
 }
 void da(int *r,int *sa,int n,int m)
 {
     int i,j,p,*x=wa,*y=wb,*t;
     for(i=;i<m;i++) wq[i]=;
     for(i=;i<n;i++) wq[x[i]=r[i]]++;
     for(i=;i<m;i++) wq[i]+=wq[i-];
     for(i=n-;i>=;i--) sa[--wq[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++) wv[i]=x[y[i]];
        for(i=;i<m;i++) wq[i]=;
        for(i=;i<n;i++) wq[wv[i]]++;
        for(i=;i<m;i++) wq[i]+=wq[i-];
        for(i=n-;i>=;i--) sa[--wq[wv[i]]]=y[i];
        for(t=x,x=y,y=t,p=,x[sa[]]=,i=;i<n;i++)
           x[sa[i]]=cmp(y,sa[i-],sa[i],j)?p-:p++;
    }
    return ;
 }
 void callheight(int *r,int n)
 {
      int i,j,k=;
      for(i=;i<=n;i++) rank[sa[i]]=i;
      for(i=;i<n;i++)
         {
            if(k) k--;
            j=sa[rank[i]-];
            while(r[i+k]==r[j+k]) k++;
            height[rank[i]]=k;
         }
 }
 bool check(int n,int mid)
 {
         int num=;
      for(int i=;i<=n;i++)
      {
         if(height[i]>=mid)
         {
                 num++;
                 if(num>=k)
                 return ;
         }
         else num=;
      }
      return ;
 }
 int main()
 {
         while(scanf("%d%d",&n,&k)!=EOF)
         {
                 int Maxx=;
              for(int i=;i<n;i++)
                  {
                          scanf("%d",&s[i]);
                          Maxx=max(Maxx,s[i]);
                  }

                  s[n]=;
                  da(s,sa,n+,Maxx+);
                  callheight(s,n);

              int low=,high=n-;
              int ans=;
              while(low<=high)
              {
                int mid=(low+high)/;
                if(check(n,mid))
                {
                        ans=mid;
                        low=mid+;
                }
                else high=mid-;
              }

                 cout<<ans<<endl;

         }
         return ;

 }