夏日炎炎,小男孩 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 <= 105
1 <= costs[i] <= 105
1 <= coins <= 108
注意数据范围,题目很容易误导我们使用 01 背包(会超时),其实这题就是简单贪心,优先选择定价小的雪糕。
class Solution:
def maxIceCream(self, costs: List[int], coins: int) -> int:
costs.sort()
ans = 0
for c in costs:
if coins < c:
break
else:
ans += 1
coins -= c
return ans
class Solution {
public int maxIceCream(int[] costs, int coins) {
Arrays.sort(costs);
int ans = 0, n = costs.length;
for (int i = 0; i < n && coins >= costs[i]; i++) {
ans++;
coins -= costs[i];
}
return ans;
}
}
class Solution {
public:
int maxIceCream(vector<int>& costs, int coins) {
sort(costs.begin(), costs.end());
int ans = 0;
for (int i = 0; i < costs.size() && coins >= costs[i]; ++i)
{
++ans;
coins -= costs[i];
}
return ans;
}
};
func maxIceCream(costs []int, coins int) int {
sort.Ints(costs)
n := len(costs)
ans := 0
for i := 0; i < n && coins >= costs[i]; i++ {
ans++
coins -= costs[i]
}
return ans
}