本题的二分和往常的二分的不太一样,我是做的了leetcode378. 有序矩阵中第K小的元素这个题才对这个题很快就有了思路。
这个题分的是值域的的范围,我们把答案定一个范围,在这个范围内进行二分。
本题中所求的运载能力值域最小是weights中的最大值,原因自己想,这个就是初始的左边界l。
最大就是weights的所有值的和,这个就是初始的有边界r。
ok现在在[l,r]二分就可以,计算当运载能力是mid的时候所需要天数day
- 若day > D 说明运载能力小了,l = mid+1
- 若day <= D,说明运载能力可以,但是不一定符合最小的要求,所以这里r = mid 而不是 r = mid -1
直到区间缩小到l == r的时候这个就是答案了。
ac代码如下
class Solution {
public:
int shipWithinDays(vector<int>& weights, int D) {
int l = weights[0],r=weights[0],mid,size = weights.size();
int temp=0,day =0;
for(int i=1;i<size;i++){
l = max(l,weights[i]);
r += weights[i];
}
while(l<r){
mid = (l+r)/2;
temp = 0;day = 0;
for(int i=0;i<size;i++){
if(temp + weights[i]> mid){
day++;temp=weights[i];
}
else temp+=weights[i];
}
if(day+1<=D) r = mid;
else l= mid+1;
}
return l;
}
};