POJ3261-Milk Patterns-后缀数组

可重叠重复k次的最长子串长度。

还是使用二分答案对heigh数组分组的做法。

 #include <cstdio>
 #include <algorithm>
 #include <cstring>
 #include <vector>
 #include <iostream>

 using namespace std;
 const int maxn = 2e4+;

 int s[maxn];
 int sa[maxn],rank[maxn],height[maxn];

 int t[maxn],t2[maxn],c[maxn];
 void build(int n,int m)
 {
     int i,j,p,*x = t, *y = t2;

     for(i=;i<m;i++)c[i]=;
     for(i=;i<n;i++)c[x[i]=s[i]]++;
     for(i=;i<m;i++)c[i]+=c[i-];
     for(i=n-;i>=;i--)sa[--c[x[i]]]=i;
     for(j=;j<=n;j<<=)
     {
         p=;
         for(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<m;i++)c[i]=;
         for(i=;i<n;i++)c[x[y[i]]]++;
         for(i=;i<m;i++)c[i]+=c[i-];
         for(i=n-;i>=;i--)sa[--c[x[y[i]]]]=y[i];
         swap(x,y);
         p=;x[sa[]]=;
         for(i=;i<n;i++)
             x[sa[i]]=y[sa[i-]]==y[sa[i]]&&y[sa[i-]+j]==y[sa[i]+j]?p-:p++;
         if(p>=n)break;
         m=p;
     }
     n--;
     int k = ;
     for(int i=;i<=n;i++) rank[sa[i] ] = i;
     for(int i=;i<n;i++)
     {
         if(k) k--;
         int j = sa[rank[i]-];
         while(s[i+k]==s[j+k]) k++;
         height[rank[i] ] = k;
     }
 }

 int N,K;

 bool check(int len,int k)
 {
     int cnt = ;
     for(int i=;i<=N;i++)
     {
         if(height[i] >= len)
         {
             cnt++;
             if(cnt >= k) return true;
         }
         else
             cnt = ;
     }
     return false;
 }

 int main()
 {
     while(~scanf("%d%d",&N,&K))
     {
         int Max = ;
         for(int i=,tmp;i<N;i++)
         {
             scanf("%d",&s[i]);
             Max = max(Max,s[i]);
         }
         s[N] = ;
         build(N+,Max+);

         int l=,r=N,mid;
         int ans = ;

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

POJ