题目
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
解法
当一拿到此题,我们第一个想法就是,一次遍历,并且同时兼顾两边来作出此题,当时当我们尝试过后,就会发现并不可行。
那为什么这样的方法不行呢?
因为当我们只从一边去我们很难兼顾两边,必然会导致顾此失彼。
只从一边出发并且要兼顾两边时,面对以下情况会就会出现难以得到正确答案:
ratings:[29,51,87,87,72,12]
从左边出发:
如果我们去判断第二个孩子的糖果数量时,它要大于第一个,小于第三个,此时第三的孩子的糖果数量都是未知的,我们很难去判断。
从右边出发也同理。
所以此时我们换个思路:是否我们一次处理一边会更好?
从左边出发并只处理左边的情况:
将最左边的糖果数量设为1;
ratings : [ 29, 51, 87, 87, 72, 12]
candysleft: [1 , 2 , 3 , 1 , 1 , 1 ]
解释:
第一次比较:51>29,所以51的糖果比29的多一个;
第二次比较:87>51,所以87的糖果比51的多一个;
第三次比较:87=87,所以第二个87的糖果为1;
第四次比较:72<82,所以72的糖果为1;
第五次比较:12<72,所以12的糖果为1;
代码实现
class Solution {
public int candy(int[] ratings) {
int ratingsLength=ratings.length;
//对比左边分数所分发的糖果
int[] candys1=new int[ratingsLength];
//对比右边的分数所分发的糖果
int[] candys2=new int[ratingsLength];
//将第一个设置为1,因为第一个没有左边
candys1[0]=1;
//将最后一个设置为1,因为最后一个没有右边
candys2[ratingsLength-1]=1;
//记录最少糖果数量
int count=0;
for(int i=1;i<ratingsLength;i++){
//当比左边大时,糖果数量为左边的数量加一,否则为1
candys1[i]=ratings[i]>ratings[i-1]?candys1[i-1]+1:1;
}
for(int i=ratingsLength-2;i>=0;i--){
//当比右边大时,糖果数量为右边的数量加一,否则为1
candys2[i]=(ratings[i]>ratings[i+1])?candys2[i+1]+1:1;
}
//当拥有了左右两边都糖果分发数量后,我们就可以就出所需的糖果
for(int i=0;i<ratingsLength;i++){
//将candys1作为结果数组,两个糖果数组比较,取大的数量
candys1[i]=candys1[i]>candys2[i]?candys1[i]:candys2[i];
count+=candys1[i];
}
return count;
}
}