CF985E Pencils and Boxes

思路：

先对a数组排序，然后使用动态规划。dp[i]表示前i个能否正确划分。则如果存在dp[j] == 1, i - j + 1 >= k并且a[i] - a[j] < d，那么dp[i] = 1，否则dp[i] = 0。可以使用树状数组对区间查询操作进行优化。复杂度O(n * log(n))。
实现：

#include <bits/stdc++.h>
using namespace std;
const int MAXN = 500005;
int n, k, d, a[MAXN], bit[MAXN];
inline int lowbit(int x) { return x & -x; }
void add(int i, int x)
{
    while (i <= n) { bit[i] += x; i += lowbit(i); }
}
int sum(int i)
{
    if (i < 0) return -1;
    if (!i) return 1;
    int ans = 0;
    while (i) { ans += bit[i]; i -= lowbit(i); }
    return ans + 1;
}
int main()
{
    while (cin >> n >> k >> d)
    {
        memset(bit, 0, sizeof bit);
        for (int i = 1; i <= n; i++) cin >> a[i];
        sort(a + 1, a + n + 1);
        for (int i = 1; i <= n; i++)
        {
            int pos = lower_bound(a + 1, a + n + 1, a[i] - d) - a;
            if (i - pos + 1 < k) continue;
            if (sum(i - k) - sum(pos - 2) >= 1) add(i, 1);
        }
        cout << (sum(n) - sum(n - 1) ? "YES" : "NO") << endl;
    }
    return 0;
}