CodeForces 985 E Pencils and Boxes

Pencils and Boxes

题意： n支铅笔，每只笔有一个颜色值， 无限个盒子， 如果一个盒子里面放笔的话， 就至少需要k支笔装在一起，并且一个盒子里面的笔的颜色值只差 不能大于 d。如果满足条件就输出YES，不满足就输出NO

题解：

对于这一条直线来说， B,C的长度为K+1， && val[C] - vla[A+1] <= d 如果我们说点C能够和前面的成员塞到一个盒子里面能成功的话， 那么[A,B]之间至少有一个点在前面处理过程中是可以分配的，设这个点为D， 那么[0,D], [D+1,C]都是合法的了。 

代码：

#include<bits/stdc++.h>
using namespace std;
#define Fopen freopen("_in.txt","r",stdin); freopen("_out.txt","w",stdout);
#define LL long long
#define ULL unsigned LL
#define fi first
#define se second
#define pb push_back
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define max3(a,b,c) max(a,max(b,c))
#define min3(a,b,c) min(a,min(b,c))
typedef pair<int,int> pll;
const int INF = 0x3f3f3f3f;
const LL mod = 1e9+7;
const int N = 5e5+100;
int a[N];
int vis[N];
int main(){
    ///Fopen;
    int n, k, d;
    scanf("%d%d%d", &n, &k, &d);
    for(int i = 1; i <= n; i++) scanf("%d", &a[i]);
    sort(a+1, a+1+n);
    vis[0] = 1;
    int pre  = 0, last = 0;
    int cnt = 1;
    for(int i = k; i <= n; i++){
        while(pre <= last && a[i] - a[pre+1] > d){
            if(vis[pre]) cnt--;
            pre++;
        }
        if(cnt > 0) vis[i] = 1;
        last++;
        if(vis[last]) cnt++;
    }
    if(vis[n]) printf("YES
");
    else printf("NO
");
    return 0;
}