题目大意:求可重复的出现k次的串的最大长度
思路:这都是套路题了,二分长度L,按照height值分组,在同一组内的组数如果大于K,则成立。PS:a[i]<=20000 本来以为基数排序为T,结果1A,也是醉了
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fod(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=2e4+10;
int n,k,v[N],sa[2][N],rk[2][N],p=1,q,a[N],K,h[N];
void calsa(int sa[N],int rk[N],int SA[N],int RK[N])
{
for(int i=1;i<=n;i++)v[rk[sa[i]]]=i;
for(int i=n;i;i--)
if(sa[i]>k)
SA[v[rk[sa[i]-k]]--]=sa[i]-k;
for(int i=n-k+1;i<=n;i++)SA[v[rk[i]]--]=i;
for(int i=1;i<=n;i++)
RK[SA[i]]=RK[SA[i-1]]+(rk[SA[i-1]]!=rk[SA[i]]||rk[SA[i]+k]!=rk[SA[i-1]+k]);
}
void getsa()
{
memset(v,0,sizeof(v));
p=0,q=1;
for(int i=1;i<=n;i++)v[a[i]]++;
for(int i=1;i<=20000;i++)v[i]+=v[i-1];
for(int i=1;i<=n;i++)
sa[p][v[a[i]]--]=i;
for(int i=1;i<=n;i++)
rk[p][sa[p][i]]=rk[p][sa[p][i-1]]+(a[sa[p][i-1]]!=a[sa[p][i]]);
for(k=1;k<n;k<<=1,swap(p,q))
calsa(sa[p],rk[p],sa[q],rk[q]);
}
void geth()
{
k=0;
for(int i=1;i<=n;i++)
if(rk[p][i]==1)h[rk[p][i]]=0;
else
{
int j=sa[p][rk[p][i]-1];
while(a[i+k]==a[j+k])k++;
h[rk[p][i]]=k;if(k>0)k--;
}
}
bool check(int x)
{
int cnt=1;
for(int i=1;i<=n;i++) {
if(h[i]>=x) {
cnt++;
if(cnt>=K) return 1;
}
else cnt=1;
}
if(cnt>=K) return 1;
return 0;
}
void solve()
{
int l=0,r=n/2,mid,ans=0;
while(l<=r) {
mid=(l+r)>>1;
if(check(mid)) ans=mid,l=mid+1;
else r=mid-1;
}
printf("%d\n",ans);
}
int main()
{
while(scanf("%d%d",&n,&K)!=EOF) {
fo(i,1,n) scanf("%d",&a[i]);
getsa(); geth();
solve();
}
return 0;
}