POJ 3261 字符串上的k次覆盖问题

题目大意：

给定一个数组，求一个最大的长度的子串至少出现过k次

 

一个子串出现多次，也就是说必然存在2个子串间的前缀长度为所求的值

通过二分答案，通过线性扫一遍，去判断出现次数，也就是说每次遇见一个height[i] , 出现次数就加1，否则重置为1

 

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
const int N = 20010;
int rank[N] , sa[N] , height[N];
int wa[N] , wb[N] , wsf[N] , wv[N];
int a[N];

int cmp(int *r , int a , int b , int l)
{
    return r[a]==r[b] && r[a+l]==r[b+l];
}

void getSa(int *r , int *sa , int n , int m)
{
    int i,j,p;
    int *x=wa , *y=wb , *t;
    for(i=0 ; i<m ; i++) wsf[i]=0;
    for(i=0 ; i<n ; i++) wsf[x[i]=r[i]]++;
    for(i=1 ; i<m ; i++) wsf[i] += wsf[i-1];
    for(i=n-1 ; i>=0 ; i--) sa[--wsf[x[i]]]=i;

    p=1;
    for(j=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++) wsf[i]=0;
        for(i=0 ; i<n ; i++) wsf[wv[i]]++;
        for(i=1 ; i<m ; i++) wsf[i]+=wsf[i-1];
        for(i=n-1 ; i>=0 ; i--) sa[--wsf[wv[i]]]=y[i];

        t=x,x=y,y=t;
        x[sa[0]]=0;
        for(p=1 , i=1 ; i<n ; i++)
            x[sa[i]] = cmp(y , sa[i-1] , sa[i] , j)?p-1:p++;
    }
    return ;
}

void getHeight(int *r , int *sa , int n)
{
    for(int i=1 ; i<=n ; i++) rank[sa[i]]=i;
    int k=0;
    int j;
    for(int 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 check(int m , int n , int k)
{
    int cnt=1;
    for(int i=1 ; i<=n ; i++){
        if(height[i]>=m){
            cnt++;
         //   cout<<"here: "<<m<<" "<<sa[i]<<" "<<sa[i-1]<<endl;
            if(cnt>=k) return true;
        }
        else {cnt=1;if(cnt>=k) return true;}
    }
    return false;
}

int main()
{
  //  freopen("a.in" , "r" , stdin);
    int n,k;
    while(~scanf("%d%d" , &n , &k))
    {
        int maxn =0 ;
        for(int i=0 ; i<n ; i++){
            scanf("%d" , a+i);
            a[i]++;
            maxn = max(maxn , a[i]);
        }
        a[n]=0;
        getSa(a , sa , n+1 , maxn+1);
        getHeight(a , sa , n);

        int l=0 , r=n , ans=0;
        while(l <= r){
            int m=(l+r)>>1;
            if(check(m,n,k)){
                ans = m;
                l=m+1;
            }else r=m-1;
        }
        printf("%d\n" , ans);
    }
    return 0;
}