poj3261 Milk Patterns（后缀数组）

【题目链接】

　　http://poj.org/problem?id=3261

【题意】

　　至少出现k次的可重叠最长子串。

【思路】

二分长度+划分height，然后判断是否存在一组的数目不小于k即可。

需要注意的是：用末尾添0的方法解决n==1时的RE问题，但是在can中需要忽略height[0]且考虑height[1..n]。

【代码】

 #include<cstdio>
 #include<cstring>
 #include<iostream>
 #define FOR(a,b,c) for(int a=(b);a<=(c);a++)
 using namespace std;

 const int maxn = +;
 const int maxm = +;

 int s[maxn];
 int sa[maxn],c[maxm],t[maxn],t2[maxn];

 void build_sa(int m,int n) {
     int i,*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(int k=;k<=n;k<<=) {
         int p=;
         for(i=n-k;i<n;i++) y[p++]=i;
         for(i=;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;

         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]+k]==y[sa[i-]+k]?p-:p++;
         if(p>=n) break;
         m=p;
     }
 }
 int rank[maxn],height[maxn];
 void getHeight(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(s[j+k]==s[i+k]) k++;
         height[rank[i]]=k;
     }
 }

 int n,k;

 bool can(int limit) {
     int cnt=;                //忽略height[0] 截至height[n]
     for(int i=;i<=n;i++) {
         if(height[i]>=limit) {
             if(++cnt>=k) return true;
         }
         else cnt=;
     }
     return false;
 }

 int main() {
     scanf("%d%d",&n,&k);
     int up=;
     FOR(i,,n-) {
          scanf("%d",&s[i]);
          s[i]++;
          up=max(up,s[i]);
     }
     s[n]=;
     build_sa(up+,n);
     getHeight(n);
     int L=,R=n;
     while(L<R) {
         int M=L+(R-L+)/;
         if(can(M)) L=M;
         else R=M-;
     }
     printf("%d\n",L);
     return ;
 }