Educational Codeforces Round 44 (Rated for Div. 2)+E. Pencils and Boxes+树状数组

题目链接:E. Pencils and Boxes

题意：N 个数，要求分成任意堆，要求每一堆只要有K个，同一堆中任意数之间差值不能超过d;

题解：用树状数组。排一下序然后从后面开始找，以当前数为最小值看能否成堆，要成堆的话要求i+k,到第一个大于a[i]+d的位置之间有能够成堆的位置。（这里判断的时候树状数组一减看是否大于0就可以了）注意初始化时在n+1的位置加1.

#include<bits/stdc++.h>
#include <iostream>
#include <string.h>
#include <algorithm>
#include <stdio.h>
#include <math.h>
#define ll long long
#define pb push_back
using namespace std;
const int N=800010;
const int INF=1<<30;
const int inf=0x3f3f3f3f;
const int maxn=5e5+10;
const int mod=1e9+7;
int n,k,d;
int lowbit(int x)
{
    return x&(-x);
}
int a[maxn],b[maxn];
bool v[maxn];
int add(int x)
{
    while(x<=n+1)
    {
        b[x]++;x+=lowbit(x);
    }
}
int get(int x)
{
    int ans=0;
    while(x>0)
    {
        ans+=b[x];x-=lowbit(x);
    }
    return ans;
}
int main()
{
    scanf("%d %d %d ",&n,&k,&d);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
    }
    sort(a+1,a+1+n);
    add(n+1);
    for(int i=n-k+1;i>=1;i--)
    {
        int pos=upper_bound(a+1,a+1+n,a[i]+d)-a;
        if(pos>=i+k-1&&get(pos)-get(i+k-1)>0)
        {
            v[i]=true;add(i);
        }
    }
    if(v[1])printf("YES
");
    else printf("NO
");
    return 0;
}