Leetcode刷题之candy

1、题目

        老师想给孩子们分发糖果，有 N 个孩子站成了一条直线，老师会根据每个孩子的表现，预先给他们评分。你需要按照以下要求，帮助老师给这些孩子分发糖果：每个孩子至少分配到 1 个糖果。评分更高的孩子必须比他两侧的邻位孩子获得更多的糖果。那么这样下来，老师至少需要准备多少颗糖果呢？

示例 1：

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

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

2、思路

        把所有孩子的糖果数初始化为 1；先从左往右遍历一遍，如果右边孩子的评分比左边的高，则右边孩子的糖果数更新为左边孩子的糖果数加 1；再从右往左遍历一遍，如果左边孩子的评分比右边的高，且左边孩子当前的糖果数不大于右边孩子的糖果数，则左边孩子的糖果数更新为右边孩子的糖果数加 1。通过这两次遍历，分配的糖果就可以满足题目要求了

3、代码

class Solution {
public:
    int candy(vector<int>& ratings) {
        if(ratings.size()==1){
            return 1;
        }
        else if (ratings.size()==2){
            if(ratings[0]==ratings[1]){
                return 2;
            }
            else{
                return 3;
            }
        }
        vector<int> results(ratings.size(),1);
        static int num=0;
        // 从左到右
        for(int i=0; i<ratings.size()-1; ++i){
            if(ratings[i]<ratings[i+1]){
                results[i+1] = results[i]+1;
            }
        }
        //从右往左
        for(int i=ratings.size()-1; i>0; --i){
            if(ratings[i]<ratings[i-1]){
                results[i-1] = max(results[i-1],results[i]+1);
            }
        }
        num = accumulate(results.begin(),results.end(),0);
        return num;
    }
};

4、结果

执行结果：通过 显示详情
执行用时：36 ms, 在所有 C++ 提交中击败了11.99%的用户
内存消耗：17.4 MB, 在所有 C++ 提交中击败了31.49%的用户