时间:2021-02-21
题目地址:https://leetcode-cn.com/problems/capacity-to-ship-packages-within-d-days/
题目难度:Medium
题目描述:
传送带上的包裹必须在 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
思路1:二分查找
算出运送完货物所需的时间
代码段1:通过
class Solution:
def shipWithinDays(self, weights: List[int], D: int) -> int:
left, right = min(weights), sum(weights)
def can_delivery(weights, ability, days):
if ability < max(weights):
return False
total_time, temp = 1, 0
for i in weights:
temp += i
if temp > ability:
total_time += 1
temp = i
print(f'ability{ability}==totaltime{total_time}')
return total_time <= days
while left <= right:
mid = left + (right - left) // 2
if can_delivery(weights, mid, D):
right = mid - 1
elif can_delivery(weights, mid, D) == False:
left = mid + 1
return left
总结:
- 今天大年初十,状态不错,进入了写代码的心流状态,写的挺快的,不到一个小时就ac,但是面试时估计只有20min,还得努力鸭。
- 有个优化的点,最小我一开始写的是1,但是应该是min(weights)
- 凡是用到for 循环的都可以考虑使用二分查找,最小值是左边界的二分,最大值是右边界的二分
- 有个思路是看D天能否运送完货物
def can_delivery(weights, ability, days):
i = 0
for d in range(days):
max_weight = ability
while ((max_weight - weights[i]) >= 0):
max_weight -= weights[i]
i += 1
if i == len(weights):
return True
return False