问题(简单)
一家商店正在打折销售糖果。每购买 两个 糖果,商店会 免费 送一个糖果。
免费送的糖果唯一的限制是:它的价格需要小于等于购买的两个糖果价格的 较小值 。
比方说,总共有
4
4
4 个糖果,价格分别为
1
,
2
,
3
和
4
1 ,2 ,3 和 4
1,2,3和4 ,一位顾客买了价格为
2
和
3
2 和 3
2和3 的糖果,那么他可以免费获得价格为
1
1
1 的糖果,但不能获得价格为
4
4
4 的糖果。
给你一个下标从
0
0
0 开始的整数数组 cost
,其中 cost[i]
表示第 i
个糖果的价格,请你返回获得 所有 糖果的 最小 总开销。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/minimum-cost-of-buying-candies-with-discount
示例 1:
输入:cost = [1,2,3]
输出:5
解释:我们购买价格为 2 和 3 的糖果,然后免费获得价格为 1 的糖果。
总开销为 2 + 3 = 5 。这是开销最小的 唯一 方案。
注意,我们不能购买价格为 1 和 3 的糖果,并免费获得价格为 2 的糖果。
这是因为免费糖果的价格必须小于等于购买的 2 个糖果价格的较小值。
示例 2:
输入:cost = [6,5,7,9,2,2]
输出:23
解释:最小总开销购买糖果方案为:
- 购买价格为 9 和 7 的糖果
- 免费获得价格为 6 的糖果
- 购买价格为 5 和 2 的糖果
- 免费获得价格为 2 的最后一个糖果
因此,最小总开销为 9 + 7 + 5 + 2 = 23 。
解题
思路
关键信息,买两颗糖果送一颗价格不高于其的糖果。即买最贵的和次贵的,可以节省第三贵的糖果。按照这样的思路,循环可解。
- 将数组降序排序;
- 定义费用变量;
- 循环一遍数组,出两次相加,第三次跳过;
代码实现
class Solution(object):
def minimumCost(self, cost):
"""
:type cost: List[int]
:rtype: int
"""
cost.sort(reverse=True)
temp = 0
for i in range(len(cost)):
if (i+1)%3!=0:
temp = temp + cost[i]
return temp
循环部分优化一下:
class Solution(object):
def minimumCost(self, cost):
"""
:type cost: List[int]
:rtype: int
"""
cost.sort(reverse=True)
return sum([cost[i] if (i+1)%3!=0 else 0 for i in range(len(cost))])