【12月打卡~Leetcode每日一题】135. 分发糖果(难度:困难)

135. 分发糖果

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

首先每个孩子给1个糖果,即可满足条件1,对于条件2,执行两次遍历即可

思路:顺序遍历,满足该孩子比左边孩子拿到的多(如果评分高),逆序遍历,在满足前者的基础上,满足该孩子比右边的孩子拿到的多(如果评分高)

class Solution:
    def candy(self, ratings: List[int]) -> int:
        n = len(ratings)
        candy = [1]*n
        if n < 2:
            return n
        for i in range(1,n):
            if ratings[i]>ratings[i-1]:
                candy[i] = candy[i-1] + 1
        for i in range(n-2,-1,-1):
            if ratings[i]>ratings[i+1]:
                candy[i] = max(candy[i+1]+1,candy[i])
        return sum(candy)

时间复杂度O(n)
空间复杂度O(n)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值