題目大意
收集了N顆鑽石(N<=50,000),現在要兩個陳列架上擺放一些鑽石。
她不會把兩個大小相差K以上的鑽石同時放在一個陳列架上(差值可等於K)現在給出K,請你幫Bessie確定她最多一共可以放多少顆鑽石在這兩個陳列架上。
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=200100;
int a[maxn],l[maxn],r[maxn];
int main()
{
int n,k;
cin>>n>>k;
for (int i=1;i<=n;i++)
cin>>a[i];
sort(a+1,a+1+n);
int j=1;
l[j]=1;
for (int i=2;i<=n;i++)
{
while (a[i]-a[j]>k) j++;
l[i]=max(l[i-1],i-j+1);
}
j=n;
r[j]=1;
for (int i=n-1;i>=1;i--)
{
while (a[j]-a[i]>k) j--;
r[i]=max(r[i+1],j-i+1);
}
int ans=0;
for (int i=1;i<n;i++)
ans=max(ans,l[i]+r[i+1]); //dp
cout<<ans;
}