POJ 3261 Milk Patterns(后缀数组+单调队列)

时间:2022-08-13 07:12:29

题意

找出出现k次的可重叠的最长子串的长度

题解

用后缀数组。

然后求出heigth数组。

跑单调队列就行了。找出每k个数中最小的数的最大值。就是个滑动窗口啊

(不知道为什么有人写二分,其实写啥都差不多快,可能是因为二分是一个常见的模型吧)

 #include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
const int N=;
int n,m,x[N],sa[N],c[N],k,height[N],rk[N],y[N],ans,q[N],head,tail,s[N];
void get_sa(){
for(int i=;i<=n;i++)c[x[i]=s[i]]++;
for(int i=;i<=m;i++)c[i]+=c[i-];
for(int i=n;i>=;i--)sa[c[x[i]]--]=i;
for(int k=;k<=n;k<<=){
int num=;
for(int i=n-k+;i<=n;i++)y[++num]=i;
for(int i=;i<=n;i++)if(sa[i]>k)y[++num]=sa[i]-k;
for(int i=;i<=m;i++)c[i]=;
for(int i=;i<=n;i++)c[x[i]]++;
for(int i=;i<=m;i++)c[i]+=c[i-];
for(int i=n;i>=;i--)sa[c[x[y[i]]]--]=y[i],y[i]=;
for(int i=;i<=n;i++){
swap(x[i],y[i]);
}
x[sa[]]=;num=;
for(int i=;i<=n;i++){
x[sa[i]]=(y[sa[i]]==y[sa[i-]]&&y[sa[i]+k]==y[sa[i-]+k])?num:++num;
}
if(n==num)break;
m=num;
}
}
void get_height(){
for(int i=;i<=n;i++)rk[sa[i]]=i;
int k=;
for(int i=;i<=n;i++){
if(rk[i]==)continue;
if(k)k--;
int j=sa[rk[i]-];
while(j+k<=n&&i+k<=n&&s[j+k]==s[i+k])k++;
height[rk[i]]=k;
}
}
int main(){
scanf("%d%d",&n,&k);
if(k==){
printf("%d",n);
return ;
}
m=;
for(int i=;i<=n;i++){
scanf("%d",&s[i]);
}
get_sa();
get_height();
head=;tail=;
for(int i=;i<=k;i++){
while(height[i]<=height[q[tail]]&&head<=tail)tail--;
q[++tail]=i;
}
ans=height[q[head]];
for(int i=k+;i<=n;i++){
while(q[head]<=i-k+&&head<=tail)head++;
while(height[i]<=height[q[tail]]&&head<=tail)tail--;
q[++tail]=i;
ans=max(ans,height[q[head]]);
}
printf("%d",ans);
return ;
}