力扣刷题笔记:135. 分发糖果(贪心算法,详细解析、代码注释)

题目:

老师想给孩子们分发糖果,有 N 个孩子站成了一条直线,老师会根据每个孩子的表现,预先给他们评分。

你需要按照以下要求,帮助老师给这些孩子分发糖果:

每个孩子至少分配到 1 个糖果。
相邻的孩子中,评分高的孩子必须获得更多的糖果。
那么这样下来,老师至少需要准备多少颗糖果呢?

示例 1:

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

示例 2:

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

来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/candy
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解题思路:

规则定义: 设学生 A 和学生 B 左右相邻,A 在 BB左边;
左规则: 当 ratings_B>ratings_A时,B的糖比 A的糖数量多。
右规则: 当 ratings_A>ratings_B时,A 的糖比 B 的糖数量多。
相邻的学生中,评分高的学生必须获得更多的糖果 等价于 所有学生满足左规则且满足右规则。

算法流程:
1、创建左右规则的两个发糖列表,数值全部预设为1;
2、利用左规则正向遍历一次,当 ratings_B>ratings_A时,B的糖比 A 的糖数量多1;
3、再利用右规则反向遍历一次,当 ratings_A>ratings_B时,A 的糖比 B 的糖数量多1;
4、最后取同时满足左右规则的数值即可(即左右规则列表中每个索引的较大者)。

代码(python):

class Solution: #贪心算法,左右规则遍历,取最大值
    def candy(self, ratings: List[int]) -> int:
        length = len(ratings)
        left = [1 for _ in range(length)] #先将每个人的糖都设为1
        right = left[:]      
        for i in range(1, length): #正向遍历
            if ratings[i]>ratings[i-1]: #左规则
                left[i] = left[i-1] + 1
        count = left[-1] #创建保存较大值的变量
        for i in range(length-2, -1, -1): #反向遍历
            if ratings[i]>ratings[i+1]: #右规则
                right[i] = right[i+1] + 1
            count += max(left[i], right[i]) #取同时满足左右规则的发糖数
        #print(left)
        #print(right)
        return count
  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 6
    评论
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值