Codeforces Round #317 [AimFund Thanks-Round] (Div. 2)

D. Minimization

时间复杂度 $O(nlogn+k^2)$

思路

• 我们首先可以想到，最大的情况就是 $k==n/2$ $$\sum _{i=1}^{n-k}|A[i]-A[i+k]|=a[n]-a[1]$$
• 然后我们考虑，我们如何使目标最小，我们想到的就是连着的串就是最小的，那么我们只要把它们排开就能使它的和小。
• 我们可以分成两组长度为 n / k 和 n / k + 1，为什么是这两组。因为这两组可以组成 n，也就是 i, i + k, i + 2k, i + 3k...... 一组，使得求和小
• 现在思考，如果我们遵从上面的想法，让他求和小，那么我们要怎么样才能算出 $sum$。我们发现，对于 $sum$ 我们只是把 $a[1]～a[n]$ 中的一部分断开，也就是 $a[i+1]-a[i]$ 会因为我们的分组而消失，那么我们为了让 $sum$ 小，我们就应该让断开的部分尽可能的大。
• 现在问题变成了我们从 $n$ 长度的数列中，选取两种子串排满整个数列，数列1的个数为 n % k 长度为 n / k + 1，数列2的个数为 k - n % k 长度为 n / k，如何选择使得断开的差尽可能大
• 例如 {1, 2, 3, 4, 5, 6, 7} n = 7, k = 2，数列1：{1, 2, 3, 4}，数列2：{5, 6, 7}，最终的排布是 {1, 5, 2, 6, 3, 7, 4}，断开的部分就是 $a[4]-a[3]$

细节注释

• 注释1：对于 i == 1 或 j == 1，我们其实不会有任何的断开，为了防止算入，我们令 a[0] = a[1]
• 注释2：当 i != 0 时，由前一项递推，断口的位置在 $pos$ 与 $pos+1$ 上
• 注释3：思路

AC(dp)

#include <bits/stdc++.h>
using namespace std;
const int N = 1e6 + 10;
LL dp[5010][5010];
int a[N];
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);
	int len1 = n / k, len2 = n / k + 1;
	int count1 = k - n % k, count2 = n % k;
	a[0] = a[1]; //注释1
	for (int i = 0; i <= count1; i ++ ) 
		for (int j = 0; j <= count2; j ++ ) {
			if (i) {
				int pos = (i - 1) * len1 + j * len2; //注释2
				dp[i][j] = max(dp[i][j], dp[i - 1][j] + a[pos + 1] - a[pos]);
			}
			if (j) {
				int pos = i * len1 + (j - 1) * len2;
				dp[i][j] = max(dp[i][j], dp[i][j - 1] + a[pos + 1] - a[pos]);
			}
		}
	cout << a[n] - a[1] - dp[count1][count2] << endl; //注释3
	return 0;
}