有一条高速公路,想要建设 m 个加油站,一共有 n 个可以选择建设加油站的地点(n >= m),在这 n 个可选位置中选择 m 个建加油站有很多种方式,每一种方式中,两个相邻加油站之间的距离都有一个最小值,每种情况的最小距离可能不一样,问所有情况中,这个最小距离的最大值是多少?
2 <= n, m <= 100000
1 <= a[i] < a[i + 1] <= 109 (a[i] 是加油站位置)
原题不是这样的,但是很多人并没有理解题什么意思(什么是最小距离最大值),所以我就翻译了一下,变成上边这道题。
输入第一行是 n 和 m,接下来下 n 个数是加油站的位置(保证是递增的,也就是 a[i + 1] > a[i])。比如下边例子:
Input 1: Input 1:
5 3 5 4
1 4 5 6 9 1 4 5 6 9
Output 1: Output 2:
4 (1 5 9) 2 (1 4 6 9)
网上很多答案说用 DP,dp[i][j] 表示前 i 个位置中选择 j 个最小距离最大值是多少,但是这道题的 n 和 m 最大都是十万,数组开不了这么大的,vs中开这么大数组直接编译就错了。所以不是 DP,而是二分+贪心。
二分+贪心
对于这样的最小值最大是多少,或者最大值最小是多少的问题,其实就是对于一个值 m,我们贪心地判断这个 m 满不满足条件,比如最小值最大是多少,我们判断 m 满不满足比所有数都小,满足的话,说明这个最大的最小值肯定 >= m,那就让 m 变大,否则让 m 变小,继续判断。这样就很明确了,对 m 进行二分,每次贪心地判断这个 m 可不可行。AC 100% 的代码如下:
#include <iostream>
#include <vector>
using namespace std;
bool check_dis(int remain_merge_time, const vector<int>& dis, int check_dis) {
int i = 0, pre_sum = 0;
while (i < dis.size()) {
if (dis[i] + pre_sum >= check_dis)
pre_sum = 0;
else {
pre_sum += dis[i];
if (remain_merge_time == 0) return false;
--remain_merge_time;
}
++i;
}
return true;
}
int main() {
int n, m, pre, t, sum = 0;
cin >> n >> m >> pre;
vector<int> dis;
for (int i = 1; i < n; ++i) {
cin >> t;
sum += t - pre;
dis.push_back(t - pre);
pre = t;
}
long long l = 0, r = sum;
while (r >= l) {
const int mid = l + (r - l) / 2;
// 贪心地看 arr 数组能不能合并成每一段都 >= tmp 的数组,只能合并 n - m 次
const bool flag = check_dis(n - m, dis, mid);
//cout.setf(ios_base::boolalpha);
//cout << "mid = " << mid << ", flag: " << flag << endl;
if (flag) l = mid + 1;
else r = mid - 1;
}
cout << r << endl;
}
读取的时候,我把 1 4 5 6 9 这样的加油站位置,变成了 3 1 1 3 这样的距离数组,那么要做的就是每次看一个 mid 能不能满足在 n - m 次合并内所有数都 >= m(为什么可以这样,是因为最左边和最右边的加油站一定会选,至于为什么一定会选,想一下就知道了)。比如 n = 5, m = 4,那就是只能合并一次,那就让 1 和 1 合并成为 2,mid 最大就是 2,如果是 n = 5, m = 3,那就可以合并两次,让 3 1 合并成 4,1 3 也合并成 4,mid 最大就是 4。
主要步骤就是分为两步:
1、二分:二分过程很简单,下界是 0 或 1 都行,上界其实不是很重要,选一个大点的就行;
2、贪心:贪心的过程也很容易理解,就是当一个数 < m 的时候,就用掉一次合并,跟它后边的合并,还小就还合并直到 >= m,合并次数为 0 后还需要合并,就说明这个 mid 选大了,如果这个 mid 返回 true,说明可以,让 mid 变大再试试,最后 r < l 的时候跳出 while。比如 2 可以,4 不可以,3 可以,那么 l 从 3 变成 4,r 还是 3,跳出循环,输出 r,也就是3;如果 2 可以,4 不可以,3 不可以,那么 l 还是 3 不变,r 变成 2,跳出 while,输出 r 也就是 2。
代码中的输出取消注释后,试一下例子1,可以看到:
![](https://i-blog.csdnimg.cn/blog_migrate/6ca50e9ae60edf45f4027829cdfe2a54.png)