传送带上的包裹必须在 D 天内从一个港口运送到另一个港口。
传送带上的第 i 个包裹的重量为 weights[i]。每一天,我们都会按给出重量的顺序往传送带上装载包裹。我们装载的重量不会超过船的最大运载重量。
返回能在 D 天内将传送带上的所有包裹送达的船的最低运载能力。
示例 1:
输入:weights = [1,2,3,4,5,6,7,8,9,10], D = 5
输出:15
解释:
船舶最低载重 15 就能够在 5 天内送达所有包裹,如下所示:
第 1 天:1, 2, 3, 4, 5
第 2 天:6, 7
第 3 天:8
第 4 天:9
第 5 天:10
请注意,货物必须按照给定的顺序装运,因此使用载重能力为 14 的船舶并将包装分成 (2, 3, 4, 5), (1, 6, 7), (8), (9), (10) 是不允许的。
示例 2:
输入:weights = [3,2,2,4,1,4], D = 3
输出:6
解释:
船舶最低载重 6 就能够在 3 天内送达所有包裹,如下所示:
第 1 天:3, 2
第 2 天:2, 4
第 3 天:1, 4
示例 3:
输入:weights = [1,2,3,1,1], D = 4
输出:3
解释:
第 1 天:1
第 2 天:2
第 3 天:3
第 4 天:1, 1
提示:
1 <= D <= weights.length <= 50000
1 <= weights[i] <= 500
看网上 题解 感觉二分挺好的,对其进行改进一下。效果还不错。
注意二分的边界条件呀。~~
bool judge(vector<int>& weights, int D, int trs)
{
int count = 0;
int trs_count = 0;
for (auto weight : weights)
{
if (trs_count + weight > trs)
{
count++;
trs_count = weight;
}
else
{
trs_count += weight;
}
}
count++;
return (count <= D) ? true : false;
}
int shipWithinDays(vector<int>& weights, int D)
{
int sum = 0, weight_max=0;
for (auto weight : weights)
{
sum += weight;
weight_max = max(weight_max, weight);
}
int L = max(weight_max, int(ceil(sum / D))), R = sum;
while (L <= R)
{
int mid = (L + R) / 2;
if (judge(weights, D, mid))
R = mid - 1;
else
L = mid + 1;
}
return L;
}