135 分发糖果【困难】

leetcode 135.分发糖果

题目链接

https://leetcode.cn/problems/candy/

题目描述

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

你需要按照以下要求，给这些孩子分发糖果：
每个孩子至少分配到 1 个糖果。 相邻两个孩子评分更高的孩子会获得更多的糖果。 请你给每个孩子分发糖果，计算并返回需要准备的最少糖果数目。
补充：如果一个孩子的评分比自己身旁的一个孩子要高，那么这个孩子就必须得到比身旁孩子更多的糖果

示例

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

解题思路

本题依旧使用的是贪心策略，455分发饼干用到了排序，本题使用两次遍历即可。先将所有孩子糖果初始化为1，第一次从左往右遍历，若右边孩子评分比左边孩子高，则将他的糖果改为左边孩子糖果数加1；第二遍从右往左，若左边孩子的评分比右边孩子高并且他的糖果不大于右边孩子，则将他的糖果改为右边孩子糖果数加1

C++ code

class Solution {
public:
    int candy(vector<int>& ratings) {
        int size = ratings.size();      //只有一个孩子时，直接返回1个糖果
        if (size < 2) {
            return size;
        }

        vector<int> num(size, 1);
        for (int i = 1; i < size; ++i) {        //从左往右遍历
            if (ratings[i] > ratings[i-1]) {
                num[i] = num[i-1] + 1;      //右边评分高就将他的糖果改为左边孩子糖果数+1
            }
        }
        for (int i = size - 1; i > 0; --i) {    //从右往左遍历
            if (ratings[i] < ratings[i-1]) {
                num[i-1] = max(num[i-1], num[i] + 1);
            }
        }
    return accumulate(num.begin(), num.end(), 0); // std::accumulate 可以很方便地求和
 
    }
};

参考:谷歌高畅leetcode刷题笔记