【LeetCode】分发糖果

目录


一、题目

n 个孩子站成一排。给你一个整数数组 ratings 表示每个孩子的评分。

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

每个孩子至少分配到 1 个糖果。

相邻两个孩子评分更高的孩子会获得更多的糖果。

请你给每个孩子分发糖果,计算并返回需要准备的 最少糖果数目 。

示例 1:

输入:ratings = [1,0,2]

输出:5

解释:你可以分别给第一个、第二个、第三个孩子分发 2、1、2 颗糖果。

示例 2:

输入:ratings = [1,2,2]

输出:4

解释:你可以分别给第一个、第二个、第三个孩子分发 1、2、1 颗糖果。

第三个孩子只得到 1 颗糖果,这满足题面中的两个条件。

提示:

n == ratings.length

1 <= n <= 2 * 104

0 <= ratings[i] <= 2 * 104


二、解法

根据题目要求,两个数相邻,若不想等,那么比较大的那个数分得的糖果更多。

假想一个最终的糖果数组,我们如何检查呢,可以先统一检查左侧,再统一检查右侧。

统一检查左侧:左边比自己评分低的,那么糖果数一定要小于自己

统一检查右侧:右边...

利用left和right数组,来存储每一侧至少应该放多少个糖果,刚开始每个小孩先分一个,然后少的再补


完整代码

python 复制代码
class Solution:
    def candy(self, ratings: List[int]) -> int:
        n = len(ratings)
        left = [1] * n
        right = [1] * n
        for i in range(1, n):
            if ratings[i] > ratings[i - 1]: 
                left[i] = left[i - 1] + 1
            if ratings[n - i - 1] > ratings[n - i]:
                right[n - i - 1] = right[n - i] + 1
        res = 0
        for i in range(n):
            res += max(left[i], right[i])
        return res

相关推荐
奋进的小暄4 分钟前
贪心算法(15)(java)用最小的箭引爆气球
算法·贪心算法
Scc_hy16 分钟前
强化学习_Paper_1988_Learning to predict by the methods of temporal differences
人工智能·深度学习·算法
巷北夜未央17 分钟前
Python每日一题(14)
开发语言·python·算法
javaisC19 分钟前
c语言数据结构--------拓扑排序和逆拓扑排序(Kahn算法和DFS算法实现)
c语言·算法·深度优先
爱爬山的老虎20 分钟前
【面试经典150题】LeetCode121·买卖股票最佳时机
数据结构·算法·leetcode·面试·职场和发展
SWHL21 分钟前
rapidocr 2.x系列正式发布
算法
雾月551 小时前
LeetCode 914 卡牌分组
java·开发语言·算法·leetcode·职场和发展
想跑步的小弱鸡1 小时前
Leetcode hot 100(day 4)
算法·leetcode·职场和发展
Fantasydg1 小时前
DAY 35 leetcode 202--哈希表.快乐数
算法·leetcode·散列表
jyyyx的算法博客1 小时前
Leetcode 2337 -- 双指针 | 脑筋急转弯
算法·leetcode