Python程序：查找救援所需最小火箭数量

假设我们有一个名为weights的数字列表，它表示人们的体重，还有一个值limit，它决定了一艘火箭的重量限制。现在每艘火箭最多可以搭载两个人。我们必须找到将所有人营救到星球所需的最少火箭数量。

因此，如果输入类似于weights = [300, 400, 300]，limit = 600，则输出将为2，因为将需要一艘火箭搭载两个体重为300的人，另一艘搭载体重为400的人。

为了解决这个问题，我们将遵循以下步骤：

• 对列表weights进行排序

• cnt := 0

• 当weights非空时，执行以下操作：

  • x := 从weights中删除最后一个元素

  • 如果weights非空且weights[0] <= limit − x，则执行以下操作：

    • 从weights中删除第一个元素

  • cnt := cnt + 1

• 返回cnt

让我们看看以下实现，以便更好地理解：

示例（Python）

 在线演示

class Solution:
def solve(self, weights, limit):
   weights.sort()
   cnt = 0
   while weights:
      x = weights.pop()
      if weights and weights[0] <= limit - x:
         weights.pop(0)
      cnt += 1
   return cnt
ob = Solution()
weights = [300, 400, 300]
limit = 600
print(ob.solve(weights, limit))

输入

[300, 400, 300], 600

输出

2

Arnab Chakraborty

更新于： 2020年10月21日

141 次浏览

开启您的 职业生涯

通过完成课程获得认证

立即开始