【后缀数组】poj-3261

/* 
 * File:   main.cpp
 * Author: lenovo
 *
 * Created on 2011年10月19日, 下午12:41
*/


/* 可重叠的k次最长重复子串
 * 与POJ1743一样，先是二分答案，然后将后缀数组分成若干组
 * 不同的地方在于这里要判断的是有没有一个组的后缀个数不小于k
 * 如果有，那么就存在k个相同子串的满足条件
 * 复杂度nlogn
*/
#include "cstdio"
#include "cstring"
#define maxn 20020

const int MAX_ = 1000002;
int wa[maxn],wb[maxn],wv[maxn],ws[MAX_];//倍增算法构造后缀数组  有个要注意的地方   ws数组的大小，应该和最大值相对应
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 = 0 ; i < m ; i++)
        ws[i] = 0;
for(i = 0 ; i < n ; i++)
        ws[x[i]=r[i]]++;
for(i = 1 ; i < m ; i++)
        ws[i] += ws[i-1];
for(i = n-1 ; i >= 0 ; i--)
        sa[--ws[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++)
            ws[i] = 0;
for(i = 0 ; i < n ; i++)
            ws[wv[i]]++;
for(i = 1 ; i < m ; i++)
            ws[i] += ws[i-1];
for(i = n-1 ; i >= 0 ; i--)
            sa[--ws[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];
void calheight(int *r,int *sa,int n)
{
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;
}
int r[maxn] , sa[maxn];


int check(int n , int k , int mid)
{
int i , count = 1;
for(i = 1 ; i <= n ; i++)
    {
if(height[i] >= mid)
        {
            count++;
if(count >= k)
return 1;
        }
else
            count = 1;
    }
return 0;
}


int main(int argc, char** argv) {
int n , k;
while(scanf("%d %d",&n,&k) != EOF)
    {
int i;
for(i = 0 ; i < n ; i++)
        {
            scanf("%d",&r[i]);
            r[i]++;
        }
        r[n] = 0;
        da(r,sa,n+1,MAX_);
        calheight(r,sa,n);
int min = 1 , max = n , mid;
while(min <= max)
        {
            mid = (min+max)/2;
if(check(n,k,mid))
                min = mid+1;
else
                max = mid-1;
        }
        printf("%d\n",max);
    }
return 0;
}

转载于:https://www.cnblogs.com/zuckerTT/archive/2011/10/19/2217675.html