135. 分发糖果

最新推荐文章于 2024-10-08 20:32:22 发布

MOXI33441

最新推荐文章于 2024-10-08 20:32:22 发布

阅读量33

点赞数

文章标签：算法贪心算法

本文链接：https://blog.csdn.net/MOXI33441/article/details/130114165

版权

该文章介绍了一个使用两次贪心策略解决糖果分配的问题。首先从前向后遍历数组，确保右侧评分高的孩子得到更多糖果，然后从后向前遍历，处理左侧评分大于右侧且糖果数较少的孩子。最后返回总的糖果数。

摘要由CSDN通过智能技术生成

两次贪心：不能左右兼顾，优先处理一边再处理另一边

1. 从前向后遍历，看右侧是否大于左侧，如果是，则在左侧的基础上+1

2.从后向前遍历，看左侧是否大于右侧并且左侧的糖果量小于等于右侧，如果是，则在右侧的基础上+1

class Solution {
public:
/*两次贪心*/
    int candy(vector<int>& ratings) {
        // 注意顺序不能排序
        // 相邻评分高的孩子会获得更多的糖果，那么我们每次都按最少的糖果数给孩子
        int pre = 1;   // 记录上一个孩子的糖果数
        vector<int> ass(ratings.size(),1);   // 总糖果数,先为每个孩子派发一个糖果
        // 从前向后遍历，先考虑右侧孩子的评分是否大于左侧不做处理
        // 这样我们可以保证，所有右侧大的孩子均满足条件
        ass[0] = 1;
        for(int i=0;i<ratings.size()-1;i++){
            if(ratings[i+1]>ratings[i]){
                ass[i+1]=ass[i]+1;
            }
        }
        // 第二次从右向左遍历，考虑左侧孩子的评分大于右侧的情况，小于不做处理
        for(int i=ratings.size()-1;i>0;i--){
            if(ratings[i-1]>ratings[i]&&ass[i-1]<=ass[i]){
                ass[i-1] = ass[i]+1;
            }
        }
        return accumulate(ass.begin(),ass.end(),0);
    }
};