Educational Codeforces Round 44 (Rated for Div. 2)E

传送
题目大意：给定一个序列，要你把这个序列任意地分成若干个集合，如果有满足以下两个条件的集合就输出YES，否则NO，1）任意一个集合不少于k个元素，2）任意一个集合中任意两个数的差绝对值不超过d

题解：先排个序再dp，dp[i]=1表示第i个数（不包括第i个数）之前的所有数满足上述条件，反之不满足。
转移方程：dp[i]=1,(若dp[j]=1且，x<=j<=i-k+1,x为序列中小于等于a[i]-d的最小数的下标）
用树状数组维护前缀和即可。

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

int a[500005];
bool dp[500005];
int bit[500005];
int n,k,d;
int lowbit(int x)
{
    return x&(-x);
}
void add(int x)
{
    for(int i=x;i<=n;i+=lowbit(i))
    {
        bit[i]++;
    }
}
int ask(int x)
{
    int sum=0;
    for(int i=x;i>0;i-=lowbit(i))
    {
        sum+=bit[i];
    }
    return sum;
}
int main()
{

    memset(bit,0,sizeof bit);
    cin>>n>>k>>d;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
    }
    sort(a+1,a+1+n);
    memset(dp,false,sizeof dp);
    dp[1]=1;
    add(1);
    for(int i=1;i<=n;i++)
    {
        int l=lower_bound(a+1,a+1+n,a[i]-d)-a;
        int r=i-k+1;
        if(ask(r)-ask(l-1)>=1)
        {
            dp[i+1]=1;
            add(i+1);
        }
    }
    if(dp[n+1])cout<<"YES"<<endl;
    else cout<<"NO"<<endl;
    return 0;
}