Leetcode 135. 分发糖果(Java实现 超详细注释!)

Leetcode 135. 分发糖果

这道题考察对贪心思想的实际运用能力,读懂题目并转换为解答方案是解决问题的第一步!加了详细的注释,方便日后复习,也希望能帮到其他小伙伴,如有错误,欢迎指正!

Java实现:

class Solution {
    public int candy(int[] ratings) {
        /**
        这道题的思路:先从左往右遍历,右边的分值比左边大,那右边的糖果就多分一个;再从右往左遍历,左边的分值比右边大,那左边的糖果就多分一个;
        如果同时满足上述两个条件,那么肯定满足相邻的孩子中,评分高的孩子必须获得更多的糖果,这里是贪心的思想,所以我们把两次遍历的结果取并集,也就是每个孩子必须都得到这两种情况中更大的那个糖果数量,这样也就能保证分发的总糖果数量尽量小;
        此外,为了满足每个孩子至少分配到 1 个糖果,我们左边和右边遍历的第一个孩子都必须给一个糖果;
        */

        方法1:
        // 先初始化两个数组
        int n = ratings.length;
        int[] left_to_right = new int[n];
        int[] right_to_left = new int[n];
        // 为了满足每个孩子至少分配到 1 个糖果,我们左边和右边遍历的第一个孩子都必须给一个糖果
        left_to_right[0] = 1;
        right_to_left[n - 1] = 1;
        // 从左向右遍历,右边比左边分高就给右边的孩子多分一个糖果,否则就只给右边的孩子分一个糖果
        for (int i = 1; i < n; i++){
            if (ratings[i] > ratings[i - 1]){
                left_to_right[i] = left_to_right[i - 1] + 1;
            }else{
                left_to_right[i] = 1;
            }
        }
        // 从右向左遍历,左边比右边分高就给左边的孩子多分一个糖果,否则就只给左边的孩子分一个糖果
        for (int j = n - 2; j >= 0; j--){
            if (ratings[j] > ratings[j + 1]){
                right_to_left[j] = right_to_left[j + 1] + 1;
            }else{
                right_to_left[j] = 1;
            }
        }
        // 统计最少的总糖果数
        int sum_candy = 0;
        // 我们把两次遍历的结果取并集并累加,也就是每个孩子必须都得到这两种情况中更大的那个糖果数量,这样也就能保证分发的总糖果数量尽量小;
        for (int c = 0; c < n; c++){
            sum_candy += Math.max(left_to_right[c],right_to_left[c]);
        }
        return sum_candy;
    }


    // 方法2:和第一种方法几乎一样,只是不需要写else,用Arrays.fill提前把所有的小朋友都发一个糖
    int n = ratings.length;
        int[] left_to_right = new int[n];
        int[] right_to_left = new int[n];
        Arrays.fill(left_to_right,1);
        Arrays.fill(right_to_left,1);

        for (int i = 1; i < n; i++){
            if (ratings[i] > ratings[i - 1]){
                left_to_right[i] = left_to_right[i - 1] + 1;
            }
        }

        for (int j = n - 2; j >= 0; j--){
            if (ratings[j] > ratings[j + 1]){
                right_to_left[j] = right_to_left[j + 1] + 1;
            }
        }
        int sum_candy = 0;
        for (int c = 0; c < n; c++){
            sum_candy += Math.max(left_to_right[c],right_to_left[c]);
        }
        return sum_candy;
    }
}
上一篇:[LeetCode-135]分发糖果


下一篇:20.12.24 135. 分发糖果