2020牛客寒假算法基础集训营2 H.施魔法

链接：

　　https://ac.nowcoder.com/acm/contest/3003/H

思路：

　　先排序，再dp，dp[i]表示用掉前i个元素的最小值，转移方程 dp[i] = min{dp[j]+a[i]-a[j]},j∈[i,j-k+1], 维护dp[j]-a[j]的前缀最小值即可

代码：

#include <bits/stdc++.h>
using namespace std;
const int N = 3e5 + 7;
int dp[N], pre, a[N], n, k;
int main() {
  scanf("%d%d", &n, &k);
  for (int i = 1; i <= n; ++i) scanf("%d", a + i);
  sort(a + 1, a + 1 + n);
  pre = -a[1];
  for (int i = 1; i < k; ++i) dp[i] = 2e9;
  for (int i = k; i <= n; ++i) {
    dp[i] = pre + a[i];
    pre = min(pre, dp[i - k + 1] - a[i - k + 2]);
  }
  cout << dp[n];
  return 0;
}