分发糖果(贪心算法)

题目描述

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

题解

  • 先确定右边评分大于左边的情况
    • 只要右边评分比左边大,右边的孩子就多一个糖果。
    • 相邻的孩子中,评分高的右孩子获得比左边孩子更多的糖果
  • 确定左孩子大于右孩子的情况
    • 只要左边评分比右边大,左边的孩子就多一个糖果。
    • 此时相邻的孩子中,评分高的左孩子获得比右边孩子更多的糖果
  • 取两次tmp[i]结果的最大值,此时便可让tmp[i]比左边** tmp**[i - 1]的糖果多,也比右边** tmp**[i + 1]的糖果多** 。**

代码

cpp 复制代码
class Solution {
public:
    int candy(vector<int>& ratings) {
        int n=ratings.size();
        if(n==1 || n==0)
            return n;
        
        vector<int> tmp(n,1);
        //从前向后遍历,右边比左边大
        for(int i=1;i<n;i++)
        {
            if(ratings[i]>ratings[i-1])
                tmp[i]=tmp[i-1]+1;
        }
        
        //左边比右边大
        for(int i=n-1;i>0;i--)
        {
            if(ratings[i-1]>ratings[i])
                tmp[i-1]=max(tmp[i-1],tmp[i]+1);
        }
        
        return accumulate(tmp.begin(),tmp.end(),0);
    }
};
相关推荐
Fine姐5 分钟前
数据挖掘3.6~3.10 支持向量机—— 核化SVM
算法·支持向量机·数据挖掘
野渡拾光1 小时前
【考研408数据结构-05】 串与KMP算法:模式匹配的艺术
数据结构·考研·算法
tainshuai4 小时前
用 KNN 算法解锁分类的奥秘:从电影类型到鸢尾花开
算法·分类·数据挖掘
Coovally AI模型快速验证9 小时前
农田扫描提速37%!基于检测置信度的无人机“智能抽查”路径规划,Coovally一键加速模型落地
深度学习·算法·yolo·计算机视觉·transformer·无人机
pusue_the_sun9 小时前
数据结构:二叉树oj练习
c语言·数据结构·算法·二叉树
RaymondZhao3410 小时前
【全面推导】策略梯度算法:公式、偏差方差与进化
人工智能·深度学习·算法·机器学习·chatgpt
zhangfeng113310 小时前
DBSCAN算法详解和参数优化,基于密度的空间聚类算法,特别擅长处理不规则形状的聚类和噪声数据
算法·机器学习·聚类
啊阿狸不会拉杆11 小时前
《算法导论》第 32 章 - 字符串匹配
开发语言·c++·算法
小学生的信奥之路11 小时前
洛谷P3817题解:贪心算法解决糖果分配问题
c++·算法·贪心算法
你知道网上冲浪吗12 小时前
【原创理论】Stochastic Coupled Dyadic System (SCDS):一个用于两性关系动力学建模的随机耦合系统框架
python·算法·数学建模·数值分析