原题地址:https://ac.nowcoder.com/acm/contest/3003/H
解题思路
一道挺难想到的动态规划(原谅我智商不够,一开始做的时候想不到这个是动态规划),先对数组进行排序,极差值就会变成最后一个减第一个。
从第2k个开始,有两种选择,要么接上上一个的尾巴,要么从之前的第k个现在的这个,重新组成一个段。
最后输出dp[n]
AC代码
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 3e5 + 10;
const int INF = 1e9+1;
int dp[maxn], a[maxn];
int main()
{
int n, k;
scanf("%d%d", &n, &k);
for (int i = 1; i <= n; i++)
{
scanf("%d", a + i);
}
sort(a+1,a+n+1);
for (int i = 1; i < k; i++)
dp[i] = INF;
dp[k] = a[k] - a[1];
for (int i = k + 1; i <= n; i++)
{
dp[i] = min(dp[i-k]+a[i]-a[i-k+1],dp[i-1]-a[i-1]+a[i]);
}
cout<<dp[n];
return 0;
}