leetcode 1833 雪糕的最大数量

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

 

/**
 * 排序:快排
 *
 * @param costs
 * @param coins
 * @return
 */
public static int maxIceCream(int[] costs, int coins) {
    if (costs == null || costs.length == 0 || coins == 0) return 0;

    quickSort(costs, 0, costs.length - 1);
    int cnt = 0;
    while (cnt < costs.length && coins >= costs[cnt]) {
        coins -= costs[cnt];
        cnt++;
    }

    return cnt;
}

public static void quickSort(int[] costs, int left, int right) {
    if (costs == null || costs.length == 0 || left >= right) return;

    int p = left, q = right;
    int temp = costs[left];
    while (p < q) {
        while (p < q && costs[q] >= temp) q--;
        while (p < q && costs[p] <= temp) p++;
        if (p < q) {
            costs[p] += costs[q];
            costs[q] = costs[p] - costs[q];
            costs[p] = costs[p] - costs[q];
        }
    }
    costs[left] = costs[p];
    costs[p] = temp;

    quickSort(costs, left, p - 1);
    quickSort(costs, p + 1, right);
}

// 测试用例
public static void main(String[] args) {
    int[] costs = new int[]{1, 3, 2, 4, 1};
    int coins = 7;
    int ans = MaxIceCream.maxIceCream(costs, coins);
    System.out.println("MaxIceCream demo01 result : " + ans);

    costs = new int[]{10, 6, 8, 7, 7, 8};
    coins = 5;
    ans = MaxIceCream.maxIceCream(costs, coins);
    System.out.println("MaxIceCream demo02 result : " + ans);

    costs = new int[]{1, 6, 3, 1, 2, 5};
    coins = 20;
    ans = MaxIceCream.maxIceCream(costs, coins);
    System.out.println("MaxIceCream demo03 result : " + ans);
}