题目
传送带上的包裹必须在 days 天内从一个港口运送到另一个港口。
传送带上的第 i 个包裹的重量为 weights[i]。每一天,我们都会按给出重量(weights)的顺序往传送带上装载包裹。我们装载的重量不会超过船的最大运载重量。
返回能在 days 天内将传送带上的所有包裹送达的船的最低运载能力。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/capacity-to-ship-packages-within-d-days
思路
可以发现结果有一个范围,所以使用二分查找
代码
class Solution {
public:
int shipWithinDays(vector<int>& weights, int days) {
// 如果一个题,不是链表、二叉树、深搜、广搜、动态规划等,首先想暴力解法
// 进一步优化,使用二分查找
// 贪心、二分
// 本题的暴力解法是:依次枚举数组的最大元素到数组的和的每一个元素,使用贪心算法
// 如果小于等于days,可以
// 这种二分查找的特点在于,二叉查找枚举的变量不再是下标索引,而是具体的值
// 二分查找一定是有序的,以前的有序是下标的有序,现在是值有一个范围
// 这类二分查找解决的是最优问题
// 这类问题的解有一个范围,这个范围决定了可以使用二分查找
// 使用二分法,时间复杂度变为logn,总体时间复杂度为:O(nlogn)
int l=*max_element(weights.begin(),weights.end());
long long sum=0;
for(int i=0;i<weights.size();i++)
sum+=weights[i];
int r=sum;
long long mid;
int ans=0;
while(l<=r){
mid=(l+r)/2; // 二分
long long sub_sum=0;
int cnt=0;
for(int ele:weights){ // 贪心
if(sub_sum+ele>mid){
sub_sum=ele;
cnt++;
if(cnt>days){
cnt=-1;
break;
}
}else{
sub_sum+=ele;
}
}
if(cnt==-1)
l=mid+1;
else{
cnt++;
if(cnt>days)
l=mid+1;
else{
r=mid-1;
ans=mid;
}
}
}
return ans;
}
};