LeetCode 135. 分发糖果【贪心】

题目描述

一群孩子站成一排,每一个孩子有自己的评分。现在需要给这些孩子发糖果,规则是如果一 个孩子的评分比自己身旁的一个孩子要高,那么这个孩子就必须得到比身旁孩子更多的糖果;所有孩子至少要有一个糖果。求解最少需要多少个糖果。

解题思路

  • 把所有孩子的糖果数初始化为 1; 先从左往右遍历一遍,如果右边孩子的评分比左边的高,则右边孩子的糖果数更新为左边孩子的 糖果数加1;

  • 再从右往左遍历一遍,如果左边孩子的评分比右边的高,且左边孩子当前的糖果数 不大于右边孩子的糖果数,则左边孩子的糖果数更新为右边孩子的糖果数加1。

  • 通过这两次遍历, 分配的糖果就可以满足题目要求了。这里的贪心策略即为,在每次遍历中,只考虑并更新相邻一 侧的大小关系。

AC

class Solution {
public:
    int candy(vector<int>& ratings) {
        int size = ratings.size();
        if(size<2) return size;
        vector<int> v(size,1);
        //先从左到右遍历
        for(int i=0;i<size-1;i++){
            if(ratings[i+1]>ratings[i])
                v[i+1]=v[i]+1;
        }
        //再从右到左遍历
        for(int i=size-1;i>=1;i--){
            if(ratings[i-1]>ratings[i])
                v[i-1]=max(v[i-1],v[i]+1);
        }
        return accumulate(v.begin(),v.end(),0);
    }
};
学如逆水行舟,不进则退
原创文章 622 获赞 2443 访问量 41万+

猜你喜欢

转载自blog.csdn.net/weixin_42429718/article/details/105948441