【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

相关推荐
圣保罗的大教堂1 小时前
leetcode 2799. 统计完全子数组的数目 中等
leetcode
SsummerC1 小时前
【leetcode100】组合总和Ⅳ
数据结构·python·算法·leetcode·动态规划
YuCaiH2 小时前
数组理论基础
笔记·leetcode·c·数组
2301_807611492 小时前
77. 组合
c++·算法·leetcode·深度优先·回溯
SsummerC3 小时前
【leetcode100】零钱兑换Ⅱ
数据结构·python·算法·leetcode·动态规划
好易学·数据结构4 小时前
可视化图解算法:二叉树的最大深度(高度)
数据结构·算法·二叉树·最大高度·最大深度·二叉树高度·二叉树深度
程序员-King.4 小时前
day47—双指针-平方数之和(LeetCode-633)
算法·leetcode
阳洞洞4 小时前
leetcode 1035. Uncrossed Lines
算法·leetcode·动态规划·子序列问题
小鹿鹿啊5 小时前
C语言编程--15.四数之和
c语言·数据结构·算法
rigidwill6665 小时前
LeetCode hot 100—最长有效括号
数据结构·c++·算法·leetcode·职场和发展