135. 分发糖果-贪心算法

题目描述

力扣题目链接(opens new window)

老师想给孩子们分发糖果，有 N 个孩子站成了一条直线，老师会根据每个孩子的表现，预先给他们评分。

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

• 每个孩子至少分配到 1 个糖果。
• 相邻的孩子中，评分高的孩子必须获得更多的糖果。

那么这样下来，老师至少需要准备多少颗糖果呢？

示例 1:

• 输入: [1,0,2]
• 输出: 5
• 解释: 你可以分别给这三个孩子分发 2、1、2 颗糖果。

示例 2:

• 输入: [1,2,2]
• 输出: 4
• 解释: 你可以分别给这三个孩子分发 1、2、1 颗糖果。第三个孩子只得到 1 颗糖果，这已满足上述两个条件。

思路

两次遍历，第一次保证左孩子得分大于右孩子时分到更多的糖果，第二次保证右孩子得分大于左孩子时分到比左孩子更多的糖果，如果两边一起考虑一定会顾此失彼。

第一次从前向后遍历：此时局部最优：只要右边评分比左边大，右边的孩子就多一个糖果，全局最优：相邻的孩子中，评分高的右孩子获得比左边孩子更多的糖果。（candyVec[i] = candyVec[i - 1] + 1）

第二次从后向前遍历，ratings[i] > ratings[i+1]]时，怎样分配candyVec[i]比较难想，因为candyVec[i]既要大于右边，又要大于左边（不能比第一遍的结果小，因为有可能ratings[i-1] < ratings[i])，所以candyVec[i]只能取max(candyVec[i], candyVec[i+1]+1)。

总结：用了两次贪心的策略：

• 一次是从左到右遍历，只比较右边孩子评分比左边大的情况。
• 一次是从右到左遍历，只比较左边孩子评分比右边大的情况。

这样从局部最优推出了全局最优

Java代码

class Solution {
    public int candy(int[] ratings) {
        int[] candyVec = new int[ratings.length];
        //第一次贪心,从左向右遍历
        candyVec[0] = 1;
        for(int i = 0; i < ratings.length - 1; i++){
            if(ratings[i] < ratings[i+1]){  //右孩子大
                candyVec[i+1] = candyVec[i] + 1;
            }else{
                candyVec[i + 1] = 1;
            }
        }
        int totalCandy = candyVec[ratings.length - 1];
        //第2次贪心,从后向前遍历
        for(int i = ratings.length - 2; i >= 0; i--){
            if(ratings[i] > ratings[i+1]){  //左孩子大
                candyVec[i] = Math.max(candyVec[i], candyVec[i+1]+1);
            }
            totalCandy += candyVec[i];
        }
        return totalCandy;
    }
}