1833.雪糕的最大数量

发布时间 2023-06-13 16:48:21作者: zwyyy456

问题描述

1833. 雪糕的最大数量 (Medium)

夏日炎炎,小男孩 Tony 想买一些雪糕消消暑。

商店中新到 n 支雪糕,用长度为 n 的数组 costs 表示雪糕的定价,其中 costs[i] 表示第 i 支雪糕的现金价格。Tony 一共有 coins 现金可以用于消费,他想要买尽可能多的雪糕。

注意: Tony 可以按任意顺序购买雪糕。

给你价格数组 costs 和现金量 coins ,请你计算并返回 Tony 用 coins 现金能够买到的雪糕的 最大数量

示例 1:

输入:costs = [1,3,2,4,1], coins = 7
输出:4
解释:Tony 可以买下标为 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 。

提示:

  • costs.length == n
  • 1 <= n <= 10⁵
  • 1 <= costs[i] <= 10⁵
  • 1 <= coins <= 10⁸

解题思路

贪心,最优方案一定是每次买最便宜的雪糕,因此可以用优先队列来模拟买雪糕的过程

代码

class Solution {
  public:
    int maxIceCream(vector<int> &costs, int coins) {
        priority_queue<int, vector<int>, std::greater<int>> pq; // 小顶堆
        for (int &price : costs) {
            pq.push(price);
        }
        int cnt = 0;
        while (!pq.empty() && coins >= pq.top()) {
            cnt++;
            coins -= pq.top();
            pq.pop();
        }
        return cnt;
    }
};