【后缀数组求可重叠的k 次最长重复子串】POJ 3261

先二分答案，然后将后缀分成若干组，判断的是有没有一个组的后缀个数不小于k。如果有，那么存在k 个相同的子串满足条件，否则不存在。这个做法的时间复杂度为O(nlogn)。

这里有一个地方要注意的，就是计数每一组的时候cnt初始值为1，因为height数组是两两之间比较的，当有1个height满足后就相当于有两个后缀满足。

#define N 1000010
#define maxn 20010
int wa[maxn],wb[maxn],wv[N],wss[N];//论文模板的ws竟然跟g++里的关键字冲突！
int r[maxn],sa[maxn];
int cmp(int *r,int a,int b,int l)
{return r[a]==r[b] && r[a+l]==r[b+l];}
/*【倍增算法O(nlgn)】待排序的字符串放在r 数组中，从r[0]到r[n-1]，长度为n，且最大值小于m
  使用倍增算法前，需要保证r数组的值均大于0。然后要在原字符串后添加一个0号字符
  所以，若原串的长度为n，则实际要进行后缀数组构建的r数组的长度应该为n+1.所以调用da函数时，对应的n应为n+1.
*/
void da(int *r,int *sa,int n,int m){//n要加1
     int i,j,p,*x=wa,*y=wb,*t;
     for(i=0;i<m;i++) wss[i]=0;
     for(i=0;i<n;i++) wss[x[i]=r[i]]++;
     for(i=1;i<m;i++) wss[i]+=wss[i-1];
     for(i=n-1;i>=0;i--) sa[--wss[x[i]]]=i;
     for(j=1,p=1;p<n;j*=2,m=p){
         for(p=0,i=n-j;i<n;i++) y[p++]=i;
         for(i=0;i<n;i++) if(sa[i]>=j) y[p++]=sa[i]-j;
         for(i=0;i<n;i++) wv[i]=x[y[i]];
         for(i=0;i<m;i++) wss[i]=0;
         for(i=0;i<n;i++) wss[wv[i]]++;
         for(i=1;i<m;i++) wss[i]+=wss[i-1];
         for(i=n-1;i>=0;i--) sa[--wss[wv[i]]]=y[i];
         for(t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1;i<n;i++)
         x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
     }
     return;
}
int rank[maxn],height[maxn];//rank[i]：i排第几；sa[i]：排第i的后缀串在哪里，互为逆运算

void calheight(int *r,int *sa,int n){//n不用加1
     int i,j,k=0;
     for(i=1;i<=n;i++) rank[sa[i]]=i;
     for(i=0;i<n;height[rank[i++]]=k){
        for(k?k--:0,j=sa[rank[i]-1];r[i+k]==r[j+k];k++);
     }
     return;
}
bool chk(int mid,int n,int k){
    int i,j;
    int cnt=1;//注意！
    for(i=1;i<=n;i++){
        if(height[i]>=mid){
            cnt++;
            if(cnt == k)return true;
        } else cnt=1;//注意！
    }
    return false;
}
int main(){
    int n;
    int k;
    while(scanf("%d%d",&n,&k) != -1){
        int i,j;int m = 0;
        for(i=0;i<n;i++){
            scanf("%d",&r[i]);
            r[i]++;
            m = max(m,r[i]);
        }
        r[n] = 0;
        da(r,sa,n+1,m+1);
        calheight(r,sa,n);
        int l=0,r = n,mid;
        int ans = 0;
        while(l<=r){
            mid = (l+r)>>1;//mid = 长度
            if(chk(mid,n,k)){
                ans = mid;
                l = mid+1;
            } else r = mid-1;
        }
        printf("%d\n",ans);
    }
    return 0;
}