洛谷【P4677】山区建小学

题目描述

政府在某山区修建了一条道路，恰好穿越总共 n 个村庄的每个村庄一次，没有回路或交叉，任意两个村庄只能通过这条路来往。已知任意两个相邻的村庄之间的距离为 d[i]​（d[i]​ 为正整数），其中，0 < i < n。为了提高山区的文化素质，政府又决定从 n 个村中选择 m 个村建小学。请根据给定的 n、m 以及所有相邻村庄的距离，选择在哪些村庄建小学，才使得所有村到最近小学的距离总和最小，计算最小值。

输入格式

第一行为 n 和 m，其间用空格间隔。

第二行为 n−1 个整数，依次表示从一端到另一端的相邻村庄的距离，整数之间以空格间隔。

例如

10 3
2 4 6 5 2 4 3 1 3

表示在 1010 个村庄中建 33 所学校。第 11 个村庄与第 22 个村庄距离为 22，第 22 个村庄与第 33 个村庄距离为 44，第 33 个村庄与第 44 个村庄距离为 66，...，第 99 个村庄到第 1010 个村庄的距离为 33。

输出格式

各村庄到最近学校的距离之和的最小值。

输入输出样例

输入 #1

10 2
3 1 3 1 1 1 1 1 3

输出 #1

18

说明/提示

1 ≤ m ≤ n < 5001，1 ≤ d[i] ≤ 1001。

思路

Hmm……

这道题在ybt里是递推，但这应该是个dp

看一下：

若只有1所小学，该最短距离应为1~m的中点

若有多所小学，该最短距离可以看作：

有村庄k，则设k+1~m村庄间要建1所小学，该区间最短距离为其中点，我们可建数组c来表示两个村庄间建一所小学的最短距离，那么k+1~m村庄间的最短距离可表示为c[k+1][m]，1~k村庄间要建n-1所小学，我们可建数组f来表示村庄离小学的最短距离，那么若所有村庄里小学的最短距离为f[m][n]，那么1~k村庄间可表示为f[k][n-1]则递推式可看成：

f[m][n]=f[k][n-1]+c[k+1][m]

代码

#include <bits/stdc++.h>
using namespace std;

int m, n, g[1010], f[1010][1010], c[1010][1010];
//m,n为题目所给，g数组为存储各个村庄距离，f数组为村庄离小学的最短距离，c数组为两个村庄间建一所小学的最短距离

const int o = 52554542; //单纯大数，没别的意义

int dis(int x, int y) { //求两个村庄间建一所小学的最短距离
	int mid = (x + y) / 2; //小学应建在中点最短
	int s = 0;
	for (int i = x; i <= y; ++i) //x，y村庄之间
		s += abs(g[i] - g[mid]); //因为有可能为小-大，所以取绝对值
	return s;
}

int main() {
	int m, n;
	cin >> m >> n;
	for (int i = 2; i <= m; ++i) { //从2开始循环输入
		cin >> g[i];
		g[i] = g[i - 1] + g[i]; //更新为1~i的距离
	}
	for (int i = 1; i <= m; ++i) {
		for (int j = 1; j <= m; ++j)
			c[i][j] = dis(i, j); //求两个村庄间建一所小学的最短距离
	}
	memset(f, o, sizeof(f));
	for (int i = 1; i <= m; ++i) {
		for (int j = 1; j <= n && j <= i; ++j) {
			if (j == 1)
				f[i][j] = c[1][i]; //若只有一所小学，直接取两个村庄间建一所小学的最短距离
			else {
				for (int k = 1; k <= i - 1; ++k) //循环k的取值
					f[i][j] = min(f[i][j], f[k][j - 1] + c[k + 1][i]); //递推方程
			}
		}
	}
	cout << f[m][n]; //输出结果
	return 0;
}
/*
*Code By XJJ
*2023-10-28 21:19:21
*/