Diamond Collector S

Pretty Boy Fox

于 2025-04-04 21:00:20 发布

阅读量732

点赞数 20

文章标签：算法 c++ 开发语言

本文链接：https://blog.youkuaiyun.com/2301_80065123/article/details/147002741

版权

P3143 [USACO16OPEN] Diamond Collector S

题目描述

奶牛 Bessie 一直喜欢闪闪发光的物体，她最近在业余时间开始了一项爱好——挖掘钻石！她收集了 $N$ 颗大小各不相同的钻石（ $\leq 50,000$ ），并希望将它们中的一部分放在谷仓里的两个展示柜中展示。

由于 Bessie 希望每个展示柜中的钻石大小相对接近，她决定如果两颗钻石的大小相差超过 $K$ ，就不能将它们放在同一个展示柜中（如果两颗钻石的大小相差恰好为 $K$ ，则可以将它们一起展示在同一个展示柜中）。给定 $K$ ，请帮助 Bessie 确定她可以在两个展示柜中一起展示的最大钻石数量。

输入格式

输入文件的第一行包含 $N$ 和 $K$ （ $\leq K \leq 1,000,000,000$ ）。

接下来的 $N$ 行每行包含一个整数，表示一颗钻石的大小。所有钻石的大小均为正数且不超过 $1, 000, 000, 000$ 。

输出格式

输出一个正整数，表示 Bessie 可以在两个展示柜中一起展示的最大钻石数量。

输入输出样例 #1

输入 #1

输出 #1

代码内容

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
const ll N=5e4+10;
ll a[N],c[N];//c[i]用来存从i-1向左最多可以取多少颗钻石。

int main()
{
    ll n,k;
    cin>>n>>k;
    a[n+1]=1e18;

    for(ll i=1;i<=n;i++)
        cin>>a[i];

    sort(a+1,a+n+1);

    ll r=2;//r是右指针
    ll mx=0,ans=0;//ma是第一个架子上的钻石数，ans是答案。
    for(ll i=1;i<=n;i++)
    {
        while(a[r]<=a[i]+k) r++;//r指向第一个大于a[i]+k的数。
        c[r]=max(r-i,c[r]);//c[r]是第r颗钻石（不包括r）左边可以摆的最多的钻石数。
        mx=max(mx,c[i]);//c[i]是第i颗钻石（不包括i）左边可以摆的最多的钻石数。
        ans=max(ans,mx+r-i);//ans是第i颗钻石（不包括i）左边可以摆的最多的钻石数+第r颗钻石（不包括r）左边可以摆的最多的钻石数。
    }

    cout<<ans<<endl;
    return 0;
}