数组costs里面是每根冰棒的价格,现手上有coins的钱,问最多能买多少个冰棒。
思路:
最容易想到的就是给价格从小到大排个序,先买便宜的再买贵的,就能保证买的最多。
public int maxIceCream(int[] costs, int coins) {
int n = costs.length;
int res = 0;
Arrays.sort(costs);
for(int i = 0; i < n; i++) {
if(coins < costs[i]) return res;
coins -= costs[i];
res ++;
}
return res;
}
上面的方法没问题,但是不是最快的,因为排序要O(nlogn),
有没有办法在O(n)复杂度?
可以看到的是,有相同价格的冰棒,它们可以一起处理。
统计每个价格的冰棒的数量,
把这些数量一起处理掉,就可以避免遍历很长的数组。
先记下最大的价格,然后就在0~最大价格的范围内遍历即可。
public int maxIceCream(int[] costs, int coins) {
int n = costs.length;
int maxCost = Arrays.stream(costs).max().getAsInt();
int[] cnt = new int[maxCost+1];
int res = 0;
for(int cost : costs) {
cnt[cost] ++;
}
for(int cost = 0; cost <= maxCost; cost++) {
if(cnt[cost] == 0) continue;
if(coins < cost) return res;
int cur = Math.min(cnt[cost], coins/cost);
coins -= cost * cur;
res += cur;
}
return res;
}