搬寝室 (HDU1421)

思路

这题类似于01背包问题，都是从一组数据里选出n组，只不过01背包里的体积是限制，这里次数是限制，所以设一个二维数组d[i][j]，表示扫描到第i个时已经放入了j对的最小解，因为要选差最小的，所以我们先对所有数据排序，相邻的就是差最小的，状态转移方程：d[i][j]=min(d[i-2][j-1]+(v[i]-v[j])^2,d[i-1][j]).

代码

#include <cstring>
#include <algorithm>
#include <iostream>
#include <string>
using namespace std;
#define ll long long
const int N = 2005;
const int mod = 1e9 + 7;
int p[N];
int d[N][N];
const int INF = 0x3f3f3f3f;
int main() {                     //这题类似背包，相邻的数有取和不取两种决策，
	int n;                       //那么d[i][j]=min(d[i-2][j-1]+p[i]-p[i-1]*2,d[i-1][j])
	int k;
	while (scanf_s("%d%d", &n, &k) != EOF)
	{
		for (int i = 1; i <= n; i++) {
			scanf_s("%d", &p[i]);
		}
		sort(p + 1, p + n + 1);
		for (int i = 0; i <= n; i++)
			for (int j = 0; j <= k; j++)//初始化,inf表示没有被访问过，也就是没有这个可能的
				d[i][j] = INF;
		for (int i = 0; i <= n; i++) {//初始化，0次时候就是0
			d[i][0] = 0;
		}
		for (int i = 2; i <= n; i++) {
			for (int j = 1; j * 2 <= i; j++) {
				d[i][j] = min(d[i - 2][j - 1] + (p[i] - p[i - 1])*(p[i] - p[i - 1]), d[i - 1][j]);
			}
		}
		cout << d[n][k] << "\n";
	}
}