二分搜索的简单应用——计算最优解 | Allocation | C/C++实现

问题描述

传送带依次送来了重量分别为wi(i = 0, 1,…, n - 1)的n个货物。现在要将这些货物装到k辆卡车上。每辆卡车可装载的货物数大于等于0，但货物重量总和不得超过卡车的最大运载量P。所有卡车的最大运载量P一致。请编写一个程序，输入n, k, wi求出装载全部货物所需的最大运载量P的最小值。

输入：
第一行输入整数n和k，用空格隔开。接下来n行输入n个整数wi，每个数占一行。
输出： 输出P的最小值，占1行。
限制：
1≤n≤100000
1≤k≤100000
1≤wi≤10000

输入示例

5 3
8
1
7
3
9

输出示例

10

讲解

确定最大运载量P时，首先要编写一个算法来计算k辆以内的卡车总共能装多少货物。思路很简单，只要卡车的运载量没达到P，我们就让其继续按顺序装货物，最后再计算所有卡车运载量的总和即可。这里我们以P为实参，编写一个返回可装载货物数v的函数 v = f§。这个函数的算法复杂度为O(n)。

现在我们只需要调用这个函数，让P从0开始逐渐自增，第一个让v大于等于n的P就是答案。但是，逐个检查P会使算法复杂度达到O(Pn)，结合问题规模，并不理想。

这里我们利用“P增加v也增加”（严格来说是P增加时v不会减少）的性质，用二分搜索来求P。可将算法复杂度降低至O(nlogP)。

AC代码如下

#include<iostream>
using namespace std;
#define MAX 100000 
typedef long long llong;

int n, k;
llong T[MAX];

//k辆最大运载量为P的卡车能装多少货物
int check(llong P){
	int i = 0;
	for(int j = 0; j < k; j++){
		llong s = 0;
		while(s + T[i] <= P){
			s += T[i];
			i++;
			if(i == n) return n;
		}
	}
	return i;
} 

int solve(){
	llong left = 0;
	llong right = 100000 * 10000;//货物数*1件货物的最大重量
	llong mid;
	while(right - left > 1){
		mid = (left + right) / 2;
		int v = check(mid);//mid == 检查mid == P时能装多少货物
		if(v >= n) right = mid;
		else left = mid; 
	} 
	return right;
}

int main(){
	cin>>n>>k;
	for(int i = 0; i < n; i++) cin>>T[i];
	llong ans = solve();
	cout<<ans<<endl;
}