算是一道dp
首先,排序好每一个架子上都是一段区间,然后只需要统计每个点向左向右最长延伸的区间。
所以我们预处理出每个点以左、以右最大能延伸的长度(最多能选几个差值不超过k的)
然后枚举每个点作为断点,sum取max即可
1 #include<cstdio> 2 #include<cstring> 3 #include<algorithm> 4 using namespace std; 5 const int N=50010; 6 int n,k,a[N],sum_l[N],sum_r[N]; 7 int read(){ 8 int sum=0; 9 char ch=getchar(); 10 while (ch<'0'||ch>'9') 11 ch=getchar(); 12 while (ch>='0'&&ch<='9'){ 13 sum=sum*10+ch-'0'; 14 ch=getchar(); 15 } 16 return sum; 17 } 18 int main(){ 19 int ans=0,j; 20 n=read(); 21 k=read(); 22 for (int i=1;i<=n;i++) 23 a[i]=read(); 24 sort(a+1,a+n+1); 25 sum_l[1]=1;j=1; 26 for (int i=2;i<=n;i++){ 27 while (a[i]-a[j]>k) j++; 28 sum_l[i]=max(sum_l[i-1],i-j+1); 29 } 30 sum_r[n]=1;j=n; 31 for (int i=n-1;i>=1;i--){ 32 while (a[j]-a[i]>k) j--; 33 sum_r[i]=max(sum_r[i+1],j-i+1); 34 } 35 for (int i=1;i<n;i++) 36 ans=max(ans,sum_l[i]+sum_r[i+1]); 37 printf("%d",ans); 38 return 0; 39 }