题目来源:https://leetcode-cn.com/problems/maximum-ice-cream-bars/
大致题意:给定一个雪糕的价格(0~10^5)数组,再给定一个金额,问最多能买几只雪糕。
思路:排序加贪心。先快排,然后从低到高,能买就买,直到钱不够了。简单直接,但是耗时大,复杂度O(n logn)。
优化:因为雪糕价格最大为10^5,可用空间换时间的方法。创建一个大小为100001的雪糕数组,下标对应元素值就是该下标对应雪糕的个数。然后从头开始遍历雪糕数组,能买几个买几个,和之前一样。
public int maxIceCream(int[] costs, int coins) {
int num = 0;
// 方法一,排序加贪心
// Arrays.sort(costs); // 排序
// int sum = 0;
// for (int i : costs) { // 贪心
// if ((sum+i) <= coins) {
// num++;
// sum += i;
// }
// else
// break;
// }
// 方法二:空间换时间
int[] freq = new int[100001];
for (int i : costs) { // 初始化雪糕数组
freq[i]++;
}
int sum = 0;
for (int i = 1; i < 100001 ; i++) { // 贪心
if ((sum + i) <= coins) {
while (freq[i]-- != 0) {
if ((sum + i) <= coins) {
sum += i;
num++;
}
else
break;
}
}
else
break;
}
return num;
}
可能最大的收获,就是明白了java对数组排序的方法:
- Arrays.sort(array)