LeetCode 135 Candy 分发糖果 中英解析 代码注释详细

题目来源：力扣（LeetCode）
题目链接：https://leetcode.cn/problems/candy

题干

There are n children standing in a line. Each child is assigned a rating value given in the integer array ratings.
n 个孩子站成一排。给你一个整数数组 ratings 表示每个孩子的评分。（assign 分配分发，in the integer array 在整数数组中）

You are giving candies to these children subjected to the following requirements:
你需要按照以下要求，给这些孩子分发糖果：（subjected to\abide by 都是服从于、跟随于、遵守的意思）

Each child must have at least one candy.
每个孩子至少分配到 1 个糖果。

Children with a higher rating get more candies than their neighbors.
相邻两个孩子评分更高的孩子会获得更多的糖果。（冒险分数高的会比他们的相邻孩子得到更多的糖果，综合给出的样例2，即，如果旁边的分数都不小于当前分数，则当前分数可以取任意值，当然倾向于取最小值）

Return the minimum number of candies you need to have to distribute the candies to the children.
请你给每个孩子分发糖果，计算并返回需要准备的 最少糖果数目 。

题解

我看的是题解里面靠前的讲的，先从左到右做一遍满足规则，然后从右到左做一遍在满足规则，时间复杂度O(N)。

从左到右：

1. 如果 ratings[i] > ratings[i - 1]，则 i 的糖果等于 i-1 的糖果 + 1
2. 如果 ratings[i] <= ratings[i - 1]，则可以取任意值，我们倾向于取最小值 1

从右到左时，如果糖果已经满足条件（ratings[i - 1] > ratings[i] 而且 i-1 的糖果大于 i 的糖果，此时是满足从左到右条件的，不可以修改），或者本身不需要增加糖果时（ratings[i - 1] <= ratings[i] ），我们都不做任何处理。除此之外， i-1 的糖果等于 i 的糖果 + 1 即可。

代码

class Solution {
public:
    int candy(vector<int>& ratings) {
        int len = ratings.size();
        vector<int> arr = vector<int> (len, 0);
        
        // size=1 需要特别处理
        arr[0] = 1;
        // 从左到右满足条件
        for (int i = 1; i < len; ++ i) {
            if (!arr[i - 1]) arr[i - 1] = 1;
            if (ratings[i] > ratings[i - 1]) arr[i] = arr[i - 1] + 1;
            else arr[i] = 1;
        }

        // debug 中间结果方便调试
        for (int i = 0; i < len; ++ i) {
            cout << arr[i] << " ";
        }
        cout << endl;
        
        // 从右向左也要满足条件，但是已经满足条件的不能再修改
        for (int i = len - 1; i >= 1; i --) {
            if (ratings[i - 1] > ratings[i] & arr[i - 1] <= arr[i]) arr[i - 1] = arr[i] + 1;
        }
        
        // 统计求和
        int res = 0;
        for (int i = 0; i < len; ++ i) {
            res += arr[i];
            cout << arr[i] << " ";
        }
        return res;
    }
};