问题描述

夏日炎炎,小男孩 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

解决思路

虽然这是一道中等难度题,但是,没啥难的,对吧~~~

代码

class Solution {
    public int maxIceCream(int[] costs, int coins) {
        int num = 0;
        Arrays.sort(costs);
        for (int i=0; i<costs.length; i++) {
            if ((coins-costs[i])>=0) {
                coins -= costs[i];
                num++;
            }
        }
        return num;
    }
}

更进一步

其实刚开始的时候,笔者没看清题目,以为是要返回所有可以买下的雪糕的序号,所以纠结了一会儿。但其实也很简单,重点就在于选择一个允许重复且有序的数据结构来存储这些雪糕,自己设计一个数据结构也行,笔者是用的是Java自带的工具类优先队列java.util.PriorityQueue,当然,我们还需要设计一个数据结构来封装雪糕数据,其中包含雪糕序号和价格,实现Comparable接口

/**
  * 封装雪糕信息
  */
class Item implements Comparable {

        private int index;
        private int value;

        Item(int index, int value) {
            this.index = index;
            this.value = value;
        }

        @Override
        public int compareTo(Object o) {
            Item other = (Item) o;
            return this.value - other.value;
        }

        @Override
        public String toString() {
            return "Item [index=" + index + ", value=" + value + "]";
        }
    }
。。。
// 添加数据
Queue<Item> items = new PriorityQueue<>();
for (int i=0; i<costs.length; i++) {
    Item item = new Item(i, costs[i]);
    items.add(item);
}
。。。