poj 3261 后缀数组

二分+后缀数组

 

#include <iostream>

#include <cstdio>

#include <cstring>

#include <cmath>

using namespace std;

const int maxn=2e4+509;

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

int r[maxn];

inline bool 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 n,int m)

{

    int wx[maxn],wy[maxn],cnt[maxn];

    int i,l,p,*x=wx,*y=wy,*t;

    memset(cnt,0,sizeof(int)*(m+1));

    for(int i=1;i<=n;i++) cnt[x[i]=r[i]]++;

    for(int i=1;i<=m;i++) cnt[i]+=cnt[i-1];

    for(int i=n;i>=1;i--) sa[cnt[x[i]]--]=i;

    for(l=1,p=1;p<n;l<<=1,m=p)

    {

        for(p=1,i=n-l+1;i<=n;i++) y[p++]=i;

        for(i=1;i<=n;i++)

        if(sa[i]>l)

        y[p++]=sa[i]-l;

        memset(cnt,0,sizeof(int)*(m+1));

        for(i=1;i<=n;i++) cnt[x[i]]++;

        for(i=1;i<=m;i++) cnt[i]+=cnt[i-1];

        for(i=n;i>=1;i--) sa[cnt[x[y[i]]]--]=y[i];

        for(t=x,x=y,y=t,p=1,x[sa[1]]=1,i=2;i<=n;i++)

        if(max(sa[i-1],sa[i])+l<=n)

        x[sa[i]]=cmp(y,sa[i-1],sa[i],l)?p:++p;

        else

        x[sa[i]]=++p;

    }

    rank=x;

    int j,k=0;

	for(i=1;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 ans,int n,int k)

{

    for(int i=2;i<=n;i++)

    if(height[i]>=ans)

    {

        int j=i+1;

        while(j<=n&&height[j]>=ans) j++;

        if(j-i+1>=k) return true;

        i=j;

    }

    return false;

}





int main()

{

    int k,n;

    scanf("%d %d",&n,&k);

    int mmax=-1;

    for(int i=1;i<=n;i++)

    {

        scanf("%d",&r[i]);

        mmax=max(mmax,r[i]);

    }

    da(r,n,mmax);

    int st=0,ed=n,mid;

    while(st<ed)

    {

        mid=st+ed+1>>1;

        if(chk(mid,n,k)) st=mid;

        else ed=mid-1;

    }

    printf("%d\n",st);

}