LeetCode 135. 分发糖果

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 *
  • 0 <= ratings[i] <= 2 *

思路:

本题可采用贪心思路确定每个孩子的糖果数。

先从左向右比,最左边的孩子的糖果数为 1,如果右边的孩子的评分高于左边的孩子则给右边孩子的糖果数为左孩子的糖果数+1,若右边的孩子评分不高于左边的孩子,则给一个糖果。这样会让比左边的孩子评分高的孩子得到更多的糖果。此时也初步形成了当前糖果分配数组 candyNums 。

再从右往左比,如果左边的孩子的评分高于右边的孩子的评分,则 左边的孩子所分得的糖果数为 当前糖果数(用于保底,如果 右边的孩子的糖果数+1< 当前糖果数,则当前糖果数不变,这时可以满足之前从左往右比确定的条件,又可以满足大于右边孩子糖果数的条件)右边的孩子的糖果数+1(保证左边的孩子的糖果大于右边的孩子) 两者之间的最大值(此时既满足从左往右确定的条件,又满足从右往左确定的条件)。

代码:

class Solution {
    public int candy(int[] ratings) {
        int len = ratings.length;
        int[] candyNums = new int[len];
        candyNums[0]=1;
        for(int i=1;i<len;i++){
            if(ratings[i]>ratings[i-1]){
                //从左往右比,如果右边的孩子评分高于左边的孩子,则右边的孩子糖果数比左边的孩子多一个
                candyNums[i]=candyNums[i-1]+1;
            }
            else{
                //否则给一个糖果
                candyNums[i]=1;
            }
        }
        //从右往左比,如果左边的孩子评分高于右边的孩子,则左边的孩子分得的糖果数为
        //当前糖果数与右边孩子糖果数+1 两者之间的最大值(为了保证比左右孩子都评分高的孩子的糖果数比左右孩子都多)
        for(int i=len-2;i>=0;i--){
            if(ratings[i]>ratings[i+1]){
                candyNums[i] = Math.max(candyNums[i],candyNums[i+1]+1);
            }
        }
        int ans = 0;
        for(int x:candyNums){
            ans += x;
        }
        return ans;
    }
}

参考:代码随想录

相关推荐
LNTON羚通3 小时前
摄像机视频分析软件下载LiteAIServer视频智能分析平台玩手机打电话检测算法技术的实现
算法·目标检测·音视频·监控·视频监控
哭泣的眼泪4084 小时前
解析粗糙度仪在工业制造及材料科学和建筑工程领域的重要性
python·算法·django·virtualenv·pygame
清炒孔心菜4 小时前
每日一题 LCR 078. 合并 K 个升序链表
leetcode
Microsoft Word5 小时前
c++基础语法
开发语言·c++·算法
天才在此5 小时前
汽车加油行驶问题-动态规划算法(已在洛谷AC)
算法·动态规划
莫叫石榴姐6 小时前
数据科学与SQL:组距分组分析 | 区间分布问题
大数据·人工智能·sql·深度学习·算法·机器学习·数据挖掘
茶猫_7 小时前
力扣面试题 - 25 二进制数转字符串
c语言·算法·leetcode·职场和发展
肥猪猪爸9 小时前
使用卡尔曼滤波器估计pybullet中的机器人位置
数据结构·人工智能·python·算法·机器人·卡尔曼滤波·pybullet
readmancynn9 小时前
二分基本实现
数据结构·算法
萝卜兽编程9 小时前
优先级队列
c++·算法