题目
夏日炎炎,小男孩 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 。
解法
方法一: dp会超时
class Solution {
public int maxIceCream(int[] costs, int coins) {
int[][] dp = new int[costs.length+1][coins+1];
for(int i = 1 ; i <= costs.length; ++i){
int c = costs[i-1];
for(int j = 0; j <= coins; ++j){
dp[i][j] = dp[i-1][j];
}
if(c > coins) {
continue;
}
for(int j = 0; j <= coins - c; ++j){
dp[i][j] = Math.max(dp[i-1][j+c]+1, dp[i][j]);
}
}
return dp[costs.length][0];
}
}
方法二: 排序+贪心 取最小值
假设购买最便宜的雪糕,在总价格不超过 coins 的情况下最多可以购买 k 支雪糕。如果将 k 支最便宜的雪糕中的任意一支雪糕替换成另一支雪糕,则替换后的雪糕的价格大于或等于替换前的雪糕的价格,因此替换后的总价格大于或等于替换前的总价格,允许购买的雪糕数量不可能超过 k。因此可以买到的雪糕的最大数量为 k。
调用api
class Solution {
public int maxIceCream(int[] costs, int coins) {
Arrays.sort(costs);
int count = 0;
int n = costs.length;
for (int i = 0; i < n; i++) {
int cost = costs[i];
if (coins >= cost) {
coins -= cost;
count++;
} else {
break;
}
}
return count;
}
}
计数排序+贪心
class Solution {
public int maxIceCream(int[] costs, int coins) {
int[] freq = new int[100001];
for (int cost : costs) {
freq[cost]++;
}
int count = 0;
for (int i = 1; i <= 100000; i++) {
if (coins >= i) {
int curCount = Math.min(freq[i], coins / i);
count += curCount;
coins -= i * curCount;
} else {
break;
}
}
return count;
}
}