P3143 [USACO16OPEN]Diamond Collector S（贪心）

题目传送门

题意： 给你n个数字，你可以对其重新排序，选择两个不相交的区间，使得两个区间内选择的数字之和最多且两个区间的极差都不超过k。

思路： 先sort一下，用一个数字r去表示第i个数字向右伸展所能达到的最远的地方的下标，用c[r]表示这个点向左伸展的话能取到的最多的数，用一个ma维护i之前，能取得的最大价值，最后维护ans就行了。

代码：

#include<bits/stdc++.h>
#pragma GCC optimize("Ofast")
#define endl '\n'
#define null NULL
#define ls p<<1
#define rs p<<1|1
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define ll long long
//#define int long long
#define pii pair<int,int>
#define pdd pair<double,double>
#define ull unsigned long long
#define all(x) x.begin(),x.end()
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define ct cerr<<"Time elapsed:"<<1.0*clock()/CLOCKS_PER_SEC<<"s.\n";
char *fs,*ft,buf[1<<20];
#define gc() (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<20,stdin),fs==ft))?0:*fs++;
inline int read(){int x=0,f=1;char ch=gc();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=gc();}
return x*f;}
using namespace std;
const int N=5e4+5;
const int inf=0x3f3f3f3f;
const int mod=998244353;
const double eps=1e-7;
const double PI=acos(-1);
int a[N],r=1,c[N],ma,ans;
signed main()
{
    int n,k;
    cin>>n>>k;
    for(int i=1;i<=n;i++)
        cin>>a[i];
    sort(a+1,a+n+1);
    a[n+1]=2e9+10;
    for(int i=1;i<=n;i++)
    {
        while(r<=n&&a[r]<=a[i]+k)
            r++;
        c[r]=max(c[r],r-i);
        ma=max(ma,c[i]);
        ans=max(ans,ma+r-i);
    }
    cout<<ans<<endl;
}