要求
楼下水果店正在促销,你打算买些苹果,arr[i] 表示第 i 个苹果的单位重量。
你有一个购物袋,最多可以装 5000 单位重量的东西,算一算,最多可以往购物袋里装入多少苹果。
示例 1:
输入:arr = [100,200,150,1000]
输出:4
解释:所有 4 个苹果都可以装进去,因为它们的重量之和为 1450。
示例 2:
输入:arr = [900,950,800,1000,700,800]
输出:5
解释:6 个苹果的总重量超过了 5000,所以我们只能从中任选 5 个。
提示:
- 1 <= arr.length <= 10^3
- 1 <= arr[i] <= 10^3
核心代码
class Solution:
def maxNumberOfApples(self, weight: List[int]) -> int:
weight.sort()
cap = 0
ans = 0
while ans < len(weight):
if cap + weight[ans] <= 5000:
cap += weight[ans]
ans += 1
else:
break
return ans
解题思路:贪心算法,先排序,然后从最小的苹果开始取,一直取到购物袋装不下苹果为止,比较简单。