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 = ;
 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 < ) return -;
     if (!i) return ;
     int ans = ;
     while (i) { ans += bit[i]; i -= lowbit(i); }
     return ans + ;
 }
 int main()
 {
     while (cin >> n >> k >> d)
     {
         memset(bit, , sizeof bit);
         for (int i = ; i <= n; i++) cin >> a[i];
         sort(a + , a + n + );
         for (int i = ; i <= n; i++)
         {
             int pos = lower_bound(a + , a + n + , a[i] - d) - a;
             if (i - pos +  < k) continue;
             if (sum(i - k) - sum(pos - ) >= ) add(i, );
         }
         cout << (sum(n) - sum(n - ) ? "YES" : "NO") << endl;
     }
     return ;
 }