代码随想录算法训练营第三十五天|860.柠檬水找零、406.根据身高重建队列、452. 用最少数量的箭引爆气球

92 阅读2分钟

860.柠檬水找零

题目链接:860. 柠檬水找零

思路:使用集合来存储五元和十元的个数(可以使用常量存储,执行速度更快)。当遇到五元,不用找零,五元的数量加一,当遇到十元,判断是否足够找零,足够的话五元的数量减一,十元的数量加一。当遇到二十元的时候,优先判断十元能否找零,不能的话再判断五元是否足够找零。局部最优:优先找零面值大的。

class Solution {
    public boolean lemonadeChange(int[] bills) {
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < bills.length; i++) {
            if (bills[i] == 5) {
                map.put(5, map.getOrDefault(5, 0) + 1);
            } else if (bills[i] == 10) {
                if (map.getOrDefault(5, 0) > 0) {
                    map.put(10, map.getOrDefault(10, 0) + 1);
                    map.put(5, map.get(5) - 1);
                } else {
                    return false;
                }
            } else if (bills[i] == 20) {
                if (map.getOrDefault(5, 0) > 0) {
                    if (map.getOrDefault(10, 0) > 0) {
                        map.put(10, map.get(10) - 1);
                        map.put(5, map.get(5) - 1);
                    } else if (map.get(5) >= 3){
                        map.put(5, map.get(5) - 3);
                    } else {
                        return false;
                    }
                } else {
                    return false;
                }
            }
        }
        return true;
    }
}

406.根据身高重建队列

题目链接:406. 根据身高重建队列

思路:本题包含两个维度,一定是先确定其中一个维度,再确定另外一个维度。首先对身高从大到小进行排序,如果身高相同的话,k小的排在前面。然后再确定k这个维度,按照k来插入队列中,k为0就插入到队列0这个位置上。局部最优:按照k来插入队列。全局最优:完成插入操作后,队列满足题目要求。

class Solution {
    public int[][] reconstructQueue(int[][] people) {
        // 身高大的人排在前面,身高相同,k小的排在前面。
        Arrays.sort(people, (a, b) -> {
            if (a[0] == b[0]) return a[1] - b[1];
            return b[0] - a[0];
        });
        List<int[]> list = new LinkedList<>();
        // 按照k来进行插入
        for (int[] p : people) {
            list.add(p[1], p);
        }
        return list.toArray(new int[people.length][]);
    }
}

452. 用最少数量的箭引爆气球

题目链接:452. 用最少数量的箭引爆气球

思路:先对气球按照起始位置进行排序,如果气球重叠,再重叠部分的最小右边界一定需要一支箭,如果当前气球和之前气球最小右边界不重叠,则需要一支新的箭。

class Solution {
    public int findMinArrowShots(int[][] points) {
        // 先按照起始位置对数组进行排序
        Arrays.sort(points, (a, b) -> {
            return Integer.compare(a[0], b[0]);
        });
        int count = 1;
        for (int i = 1; i < points.length; i++) {
            if (points[i][0] > points[i - 1][1]) { // 气球i和i-1不挨着,增加一支箭
                count++;
            } else { // 气球i和i - 1重叠,更新最小右边界。
                points[i][1] = Math.min(points[i - 1][1], points[i][1]);
            }
        }
        return count;
    }
}