1833.雪糕的最大數(shù)量
難度:中等
題目
夏日炎炎,小男孩 Tony 想買一些雪糕消消暑。
商店中新到 n 支雪糕,用長度為 n 的數(shù)組 costs 表示雪糕的定價,其中 costs[i] 表示第 i 支雪糕的現(xiàn)金價格。Tony 一共有 coins 現(xiàn)金可以用于消費(fèi),他想要買盡可能多的雪糕。
給你價格數(shù)組 costs 和現(xiàn)金量 coins ,請你計算并返回 Tony 用 coins 現(xiàn)金能夠買到的雪糕的 最大數(shù)量 。
注意:Tony 可以按任意順序購買雪糕。
提示:
- costs.length == n
- 1 <= n <= 10 ^ 5
- 1 <= costs[i] <= 10 ^ 5
- 1 <= coins <= 10 ^ 8
示例
示例 1:
輸入:costs = [1,3,2,4,1], coins = 7
輸出:4
解釋:Tony 可以買下標(biāo)為 0、1、2、4 的雪糕,總價為 1 + 3 + 2 + 1 = 7
示例 2:
輸入:costs = [10,6,8,7,7,8], coins = 5
輸出:0
解釋:Tony 沒有足夠的錢買任何一支雪糕。
示例 3:
輸入:costs = [1,6,3,1,2,5], coins = 20
輸出:6
解釋:Tony 可以買下所有的雪糕,總價為 1 + 6 + 3 + 1 + 2 + 5 = 18 。
分析
搞不懂這種題怎么能算作是中等?思路也太過簡單了吧!
簡單排序
- 首先我們將雪糕花費(fèi),按照從大到小的方式排列
- 循環(huán)雪糕,每次錢幣減去當(dāng)前雪糕的價格
- 重復(fù)2操作,當(dāng)錢不足以買雪糕的時候,返回index - 1即可.
棧操作
- 首先我們將雪糕花費(fèi),按照逆序排列
- 然后開始while循環(huán),錢幣大于等于棧頂,則出棧,雪糕數(shù)+1
- 結(jié)束while循環(huán)后,返回結(jié)果即可...
堆排序
- 將costs轉(zhuǎn)化為堆
- 每次pop堆頂,當(dāng)錢幣小于堆頂時終止操作
解題1.簡單排序
class Solution:
def maxIceCream(self, costs, coins):
costs.sort()
ret = 0
for index, cost in enumerate(costs):
coins -= cost
if coins < 0:
break
ret += 1
return ret
解題2.棧操作
class Solution:
def maxIceCream(self, costs, coins):
costs.sort(reverse=True)
num = 0
while costs and coins >= costs[-1]:
num += 1
coins -= costs.pop()
return num
解題2.堆排序
import heapq
class Solution:
def maxIceCream(self, costs, coins):
num = 0
heapq.heapify(costs)
while costs:
coins -= heapq.heappop(costs)
if coins < 0:
return num
num += 1
return num
歡迎關(guān)注我的公眾號: 清風(fēng)Python,帶你每日學(xué)習(xí)Python算法刷題的同時,了解更多python小知識。
我的個人博客:https://qingfengpython.cn