【二分+贪心】【最大值最小化】公路上建加油站,最小距离最大化

有一条高速公路,想要建设 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,可以看到:

  • 1
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值