Leetcode101_贪心算法_135发糖果

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

题解

这道题看题解是需要遍历两遍，先从左往右遍历，如果右边孩子的评分比左边的高，则右边孩子的糖果数更新为左边孩子的糖果数加 1；再从右往左遍历一遍，如果左边孩子的评分比右边的高，且左边孩子当前的糖果数不大于右边孩子的糖果数，则左边孩子的糖果数更新为右边孩子的糖果数加 1。
关于为什么要遍历两遍，我是这么理解的：
比如数组为ratings = [1 2 3 4 3 2 1]
如果只从左往右比较一次，那么只比较了右边孩子比左边孩子高的情况，最后结果为 1 2 3 4 1 1 1
因此还需要从右往左遍历一遍，看左边孩子是不是比右边孩子高，这个时候由于已经从左往右遍历一遍了，如果这个时候左边孩子比右边孩子高，而且左边孩子的糖果数本来就比右边多，那么本身就满足题目要求，这个时候不需要加一，所以第二次遍历还需要加一个判断条件才行

class Solution {
public:
    int candy(vector<int>& ratings) {
        vector <int> numbers(ratings.size(),1);
        int num = 0;
        for(int i = 1; i <ratings.size(); i++){
            if(ratings[i] > ratings[i - 1]){
                numbers[i] = numbers[i - 1] + 1;
            }
        }
        for(int j =ratings.size() - 2;j >= 0; j--){
            if(ratings[j] > ratings[j + 1] && numbers[j] <= numbers[j + 1]){
                numbers[j] = numbers[j + 1] + 1;
            }
        }
        for(int k = 0; k <ratings.size(); k++){
            num = num + numbers[k];
        }
        return num;
    }
};