题目描述
政府在某山区修建了一条道路,恰好穿越总共 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
*/