力扣 135. 分发糖果

一、题目描述

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

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

• 每个孩子至少分配到 1 个糖果。
• 相邻两个孩子评分更高的孩子会获得更多的糖果。
• 请你给每个孩子分发糖果，计算并返回需要准备的最少糖果数目。

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

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

---

二、题解

通过贪心法求解，第一轮处理右边分更高的情况，第二轮处理左边分更高的情况，时间复杂度和空间复杂度均为 $O(n)$：

class Solution {
public:
    int candy(vector<int> &ratings) {
        vector<int> records(ratings.size(), 1); // 用于记录每个孩子的最少糖果数目
        int ret = 0; // 最少糖果数目

        /* 首先保证右边分更高的孩子糖果更多 */
        for (int i = 1; i < ratings.size(); i++) {
            if (ratings.at(i) > ratings.at(i - 1)) {
                records.at(i) = records.at(i - 1) + 1;
            }
        }

        /* 其次保证左边分更高的孩子糖果更多 */
        for (int i = ratings.size() - 2; i >= 0; i--) {
            if (ratings.at(i) > ratings.at(i + 1)) {
                records.at(i) = max(records.at(i), records.at(i + 1) + 1);
            }
        }

        /* 累加糖果总数 */
        for (int i = 0; i < records.size(); i++) {
            ret += records.at(i);
        }

        return ret;
    }
};