Question

1833. 雪糕的最大数量

English Version

题目描述

夏日炎炎，小男孩 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 <= 105
• 1 <= costs[i] <= 105
• 1 <= coins <= 108

解法

方法一：贪心 + 排序

要买尽可能多的雪糕，且可以按任意顺序购买，因此，我们应该优先选择定价小的雪糕。

对数组 costs 进行排序，然后从定价最小的雪糕开始，依次购买，直到不能购买为止，返回能购买的雪糕数量。

时间复杂度 $O(n \times \log n)$，空间复杂度 $O(\log n)$。其中 $n$ 是数组 costs 的长度。

class Solution:
  def maxIceCream(self, costs: List[int], coins: int) -> int:
    costs.sort()
    for i, c in enumerate(costs):
      if coins < c:
        return i
      coins -= c
    return len(costs)

class Solution {
  public int maxIceCream(int[] costs, int coins) {
    Arrays.sort(costs);
    int n = costs.length;
    for (int i = 0; i < n; ++i) {
      if (coins < costs[i]) {
        return i;
      }
      coins -= costs[i];
    }
    return n;
  }
}

class Solution {
public:
  int maxIceCream(vector<int>& costs, int coins) {
    sort(costs.begin(), costs.end());
    int n = costs.size();
    for (int i = 0; i < n; ++i) {
      if (coins < costs[i]) return i;
      coins -= costs[i];
    }
    return n;
  }
};

func maxIceCream(costs []int, coins int) int {
  sort.Ints(costs)
  for i, c := range costs {
    if coins < c {
      return i
    }
    coins -= c
  }
  return len(costs)
}

function maxIceCream(costs: number[], coins: number): number {
  costs.sort((a, b) => a - b);
  const n = costs.length;
  for (let i = 0; i < n; ++i) {
    if (coins < costs[i]) {
      return i;
    }
    coins -= costs[i];
  }
  return n;
}

/**
 * @param {number[]} costs
 * @param {number} coins
 * @return {number}
 */
var maxIceCream = function (costs, coins) {
  costs.sort((a, b) => a - b);
  const n = costs.length;
  for (let i = 0; i < n; ++i) {
    if (coins < costs[i]) {
      return i;
    }
    coins -= costs[i];
  }
  return n;
};