给你一个整数数组 prices
,它表示一个商店里若干巧克力的价格。同时给你一个整数 money
,表示你一开始拥有的钱数。
你必须购买 恰好 两块巧克力,而且剩余的钱数必须是 非负数 。同时你想最小化购买两块巧克力的总花费。
请你返回在购买两块巧克力后,最多能剩下多少钱。如果购买任意两块巧克力都超过了你拥有的钱,请你返回 money
。注意剩余钱数必须是非负数。
示例 1:
输入:prices = [1,2,2], money = 3 输出:0 解释:分别购买价格为 1 和 2 的巧克力。你剩下 3 - 3 = 0 块钱。所以我们返回 0 。
难点:要选最便宜的两块巧克力,且剩余的钱必须是非负数(影响不大,一定会输出非负数:0或原来的money)
class Solution(object):
def buy_two_chocolates(self,prices, money):
# 首先对价格数组进行排序
prices.sort()#从低到高排序
# 检查是否有足够的钱购买至少两块巧克力
if prices[0] + prices[1] > money:
return money # 如果无法购买两块巧克力,则返回原始的钱数
# 计算购买两块巧克力后剩余的钱数
# remaining_money = money - (prices[0] + prices[1])
#
# # 剩余的钱数必须是非负数
# return max(remaining_money, 0)
else:
return money-prices[0]-prices[1]
# 计算购买两块巧克力后剩余的钱数
# 示例测试
solution=Solution()
prices = [3, 2, 2]
print(solution.buy_two_chocolates(prices, 3)) # 输出: 0
思考:文心一言给出了另一种写法
# 计算购买两块巧克力后剩余的钱数 # remaining_money = money - (prices[0] + prices[1]) # # # 剩余的钱数必须是非负数 # return max(remaining_money, 0)
这个在if循环外,但是remaining_money一定是非负数啊,因为它在if循环下面,应该是默认
prices[0] + prices[1] <=money,所以为什么要加 return max(remaining_money, 0) ?