分发糖果

732 阅读2分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第6天,点击查看活动详情

说在前面

🎈不知道大家对于算法的学习是一个怎样的心态呢?为了面试还是因为兴趣?不管是处于什么原因,算法学习需要持续保持,今天让我们一起来看看这一道题目————分发糖果,可以使用贪心思想来解答这道题目。

题目描述

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

思路分析

题目主要有下面三点要求:

  • 1、每个孩子至少有一个糖果
  • 2、相邻的孩子中,评分更高的应该有更多的糖果
  • 3、计算消耗的最少糖果数

看到这里我们就可以很明显地看出这是一道贪心的题目,我们可以分成两步来分析:

1、只考虑左边的孩子

我们可以先从左往右遍历,以第一位孩子一个糖果为基准,比较遍历到的孩子与其左边孩子的评分,评分高于左边孩子则多分其一个糖果。

image.png

2、再与右边孩子进行比较

以第一遍遍历得出的结果为基准,从右往左遍历比较,当前孩子与右边孩子进行比较,如果右边孩子评分不高于当前孩子,则当前孩子应分糖果数为Max(右边孩子糖果+1,当前孩子第一次遍历结果);如果右边孩子评分高于当前孩子,则当前孩子应分糖果数为Max(右边孩子糖果,当前孩子第一次遍历结果)

image.png

两遍遍历过后即可得出每个孩子应分得的最少糖果,具体代码如下:

AC代码

/**
 * @param {number[]} ratings
 * @return {number}
*/
var candy = function(ratings) {
 let ans = new Array(ratings.length).fill(1);
 let res = 0;
 for(let i = 1; i < ratings.length; i++){
     if(ratings[i] > ratings[i - 1]){
        ans[i] = ans[i - 1] + 1;
     }
 }
 for(let i = ratings.length - 1; i >= 0; i--){
     if(i != ratings.length - 1 && ratings[i] > ratings[i + 1]){
         ans[i] = Math.max(ans[i + 1] + 1,ans[i]);
     }
     res += ans[i];
 }
 return res;
};

说在后面

🎉这里是JYeontu,喜欢算法,GDCPC打过卡;热爱羽毛球,大运会打过酱油。毕业一年,两年前端开发经验,目前担任H5前端开发,算法业余爱好者,有空会刷刷算法题,平时喜欢打打羽毛球🏸 ,也喜欢写些东西,既为自己记录📋,也希望可以对大家有那么一丢丢的帮助,写的不好望多多谅解🙇,写错的地方望指出,定会认真改进😊,在此谢谢大家的支持,我们下文再见🙌。