VIRTUALS

the virtual labs for the virtuals

0%

LeetCode 1833. 雪糕的最大数量

摘要:
排序+贪心题。

题目

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

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

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

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

示例 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^5$
  • $1 <= costs[i] <= 10^5$
  • $1 <= coins <= 10^8$

贪心+计数排序

因为数组中数值存在一个范围,那么我们可以用计数排序的方法优化排序的复杂度为线性。

按照价格升序选择雪糕,最终个数一定是最多的。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
class Solution {
public:
static constexpr int N = 1e5 + 10;
int cnt[N];
int maxIceCream(vector<int>& costs, int coins) {
int len = costs.size();
for (int i = 0; i < len; i++) {
cnt[costs[i]]++;
}
int res = 0;
for (int i = 0; i < N && coins >= i; i++) {
for (; cnt[i]--;) {
if (coins >= i) {
res ++;
coins -= i;
}
else {
break;
}
}
}
return res;
}
};