LeetCode 135.分发糖果

题目描述

n 个孩子站成一排。给你一个整数数组 ratings 表示每个孩子的评分。

你需要按照以下要求，给这些孩子分发糖果：

• 每个孩子至少分配到 1 个糖果。
• 相邻两个孩子评分更高的孩子会获得更多的糖果。

请你给每个孩子分发糖果，计算并返回需要准备的 最少糖果数目 。

示例 1：

输入：ratings = [1,0,2]
输出：5
解释：你可以分别给第一个、第二个、第三个孩子分发 2、1、2 颗糖果。

示例 2：

输入：ratings = [1,2,2]
输出：4
解释：你可以分别给第一个、第二个、第三个孩子分发 1、2、1 颗糖果。
     第三个孩子只得到 1 颗糖果，这满足题面中的两个条件。

题解：

参考k神的解答写一下我的想法： 主要就是把题目中给出的两个条件再细化，一一处理：

一、每个孩子至少分配到 1 个糖果。

二、评分更高的孩子必须比他两侧的邻位孩子获得更多的糖果。

1. 评分比左侧孩子高的孩子，至少比左侧孩子多一颗糖

2. 评分比右侧孩子高的孩子，至少比右侧孩子多一颗糖

接下来代码中的三次遍历，就是对以上条件的依次处理

    public int candy(int[] ratings) {
        int res = 0;
        int[] candy = new int[ratings.length];

        // 初始给所有孩子一颗糖果,满足条件一
        for (int i = 0; i < candy.length; i++){
            candy[i] = 1;
        }

        // 先从左往右遍历，首先满足一个孩子比他左边孩子评分高就多糖的规则（第一个孩子可以忽略了，因为他左边没孩子）
        // 满足条件二中的第1步
        for (int i = 1; i < candy.length; i++){
            if (ratings[i] > ratings[i - 1]){   
                candy[i] = candy[i - 1] + 1;
            }
        }

        // 然后从右往左遍历，满足一个孩子比他右边孩子评分高就多糖的规则（最后一个孩子也可以忽略）
        // 满足条件二中的第2步
        for (int i = candy.length - 2; i >= 0; i--){
                // 注意这里的第二个判断：如果糖以及比左孩子多了，自然不用再考虑，此种情况只在最后一次遍历会出现，比如[1,2,3,2,1]这种例子
            if (ratings[i] > ratings[i + 1] && candy[i] <= candy[i + 1]){
                candy[i] = candy[i + 1] + 1;
            }
        }
        for (int num : candy){
            res += num;
        }

        return res;
    }