购买两块巧克力(LeetCode日记)

LeetCode-2706-购买两块巧克力

题目信息:

给你一个整数数组 prices ，它表示一个商店里若干巧克力的价格。同时给你一个整数 money ，表示你一开始拥有的钱数。

你必须购买 恰好 两块巧克力，而且剩余的钱数必须是 非负数 。同时你想最小化购买两块巧克力的总花费。

请你返回在购买两块巧克力后，最多能剩下多少钱。如果购买任意两块巧克力都超过了你拥有的钱，请你返回 money 。注意剩余钱数必须是非负数。

• 示例1：

输入：prices = [1,2,2], money = 3
输出：0
解释：分别购买价格为 1 和 2 的巧克力。你剩下 3 - 3 = 0 块钱。所以我们返回 0 。

• 示例2：

输入：prices = [3,2,3], money = 3
输出：3
解释：购买任意 2 块巧克力都会超过你拥有的钱数，所以我们返回 3 。

提示：

• $2<=prices.length<=50$
• $1<=prices[i]<=100$
• $1<=money<=100$

相关标签 ：数组、排序

题解

今天的问题属于简单难度，果然容易。我一共想出了两个方法来解决上面的问题。下面我将一一告诉大家。

方法1：排序

由题意，我们只需要找出 $price$ 数组中花费最小的两个值并与 $money$ 比较 ，如果 $money$ 大于花费最小的两数之和 ，则返回三者之间的差值，否则，则返回 $money$ 的值。
为了寻找 $price$ 数组中花费最小的两个值 ，一下子就想到了排序，将$price$ 数组排序后，取前排序后的数组前两位即可。

实现代码（Python）

class Solution:
    def buyChoco(self, prices: List[int], money: int) -> int:
        prices.sort()
        if prices[0] + prices[1] > money:
            return money
        else:
            return money - prices[0] - prices[1] 

复杂度分析：

• 时间复杂度：$O(nlo{g}_{2}n)$。排序的时间复杂度为 $nlo{g}_{2}n$ 。
• 空间复杂度：$O(1)$ 。
  

方法2：一次遍历

因为排序的时间复杂度为$O(nlo{g}_{2}n)$，为了更高效的解决问题，我们也可以使用一次遍历的方法来解决问题。
在一次遍历中，我们需不断维护最小值和次小值，遇到比最小值更小的值，则更新最小值和次小值，遇到介于最小值和次小值之间的值，则更新次小值。在遍历结束后，我们所维护的最小值和次小值即我们所寻之值。

实现代码（Python）

class Solution:
    def buyChoco(self, prices: List[int], money: int) -> int:
        cost_1 , cost_2 = inf , inf # 记录最便宜的两块巧克力的价格
        
        for cost in prices:
            if cost < cost_1 : #cost_1 记录最便宜的巧克力，
            	#更新最小值和次小值
                cost_2 = cost_1
                cost_1 = cost 
            elif cost < cost_2 : #cost_2 记录第二便宜的巧克力
            	#更新次小值
                cost_2 = cost
            else:
                pass
        if money < cost_1 + cost_2 :
            return money
        else:
            return money - cost_1 - cost_2

复杂度分析：

• 时间复杂度：$O(n)$。
• 空间复杂度：$O(1)$ 。
  

题记：

---

• 研究生在读，我会尽量保持LeetCode每日一题的思路和代码输出。希望大家多多支持。
• 水平有限，希望各位大佬能够批评指正。您的教诲是我进步的船帆。
• 希望各位跟我一样的小白能跟我一起参与到做题和讨论中来。共同进步是我所能期盼的最高愿想。
• 您的点赞和关注是我坚持分享的动力泉源，希望能将这件简单平凡的事一直做下去。感谢大家。

---