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
",st);
}