题意是求可以重复的出现次数超过k次的子串,最长是多少。
样例输入
8 2
1 2 3 2 3 2 3 1输出 4
二分验证是否有连续的k-1个height大于所验证的答案。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define SZ 20005
using namespace std;
int L[SZ],a[SZ];
int n,K;
struct Suffix
{
int sa[SZ];
int rank[SZ<<1],tmp[SZ];
int height[SZ];
int cnt[SZ],sec[SZ];
void get_sa()
{
for(int i=1;i<=n;i++) rank[i]=a[i];
for(int i=1;i<=n;i++) cnt[rank[i]]++;
for(int i=1;i<=n;i++) cnt[i]+=cnt[i-1];
for(int i=1;i<=n;i++) sa[cnt[rank[i]]--]=i;
for(int k=1;k<=n;k<<=1)
{
int top=0;
for(int i=n-k+1;i<=n;i++) sec[++top]=i;
for(int i=1;i<=n;i++) if(sa[i]>k) sec[++top]=sa[i]-k;
for(int i=0;i<=n;i++) cnt[i]=0;
for(int i=1;i<=n;i++) cnt[rank[i]]++;
for(int i=1;i<=n;i++) cnt[i]+=cnt[i-1];
for(int i=n;i>=1;i--) sa[cnt[rank[sec[i]]]--]=sec[i];
top=tmp[sa[1]]=1;
for(int i=2;i<=n;i++)
{
if(rank[sa[i]]!=rank[sa[i-1]]||rank[sa[i]+k]!=rank[sa[i-1]+k]) top++;
tmp[sa[i]]=top;
}
for(int i=1;i<=n;i++)
rank[i]=tmp[i];
}
int k=0;
for(int i=1;i<=n;i++)
{
if(k) k--;
while(i+k<=n&&sa[rank[i]-1]+k<=n&&a[i+k]==a[sa[rank[i]-1]+k]) k++;
height[rank[i]]=k;
}
}
bool check(int mid)
{
int tot=0;
for(int i=2;i<=n;i++)
{
if(height[i]>=mid) tot++;
else tot=0;
if(tot>=K-1) return true;
}
return false;
}
}SA;
int main()
{
scanf("%d%d",&n,&K);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
L[i]=a[i];
}
sort(L+1,L+n+1);
int len=unique(L+1,L+n+1)-L;
for(int i=1;i<=n;i++)
a[i]=lower_bound(L+1,L+len+1,a[i])-L;
SA.get_sa();
int l=0,r=n+1;
while(r-l>1)
{
int mid=(l+r)>>1;
if(SA.check(mid)) l=mid;
else r=mid;
}
printf("%d",l);
return 0;
}