分发糖果力扣-135
分发糖果力扣–135
题目
n
个孩子站成一排。给你一个整数数组
ratings
表示每个孩子的评分。
你需要按照以下要求,给这些孩子分发糖果:
- 每个孩子至少分配到
1
个糖果。 - 相邻两个孩子评分更高的孩子会获得更多的糖果。
请你给每个孩子分发糖果,计算并返回需要准备的 最少糖果数目 。
示例 1:
输入:ratings = [1,0,2]
输出:5
解释:你可以分别给第一个、第二个、第三个孩子分发 2、1、2 颗糖果。
示例 2:
输入:ratings = [1,2,2]
输出:4
解释:你可以分别给第一个、第二个、第三个孩子分发 1、2、1 颗糖果。
第三个孩子只得到 1 颗糖果,这满足题面中的两个条件。
思路
这道题每个数字要和左边的比较,也要和右边的比较,如果两边同时比较,会混淆
所以先从前往后,把下标为0的糖果数目先定为1,比较右边的大于左边的情况 ratings【i】>ratings[i-1],因为这里有i-1,所以遍历的时候是从下标为1的位置开始的。
ratings数组:1 2 2 5 4 3 2
糖果: 1 2 1 2 1 1 1
接着从后往前,比较左边大于右边的情况。比较右边的大于左边的情况 ratings【i】>ratings[i+1],因为这里有i+1,所以遍历的时候是从下标为ratings.length-2的位置开始的。 为什么是从后往前呢?
rating[5]与rating[4]的比较 要利用上 rating[5]与rating[6]的比较结果,所以 要从后向前遍历。
如果是从前往后,这个数组
ratings数组:1 2 2 5 4 3 2
糖果右大于左1 2 1 2 1 1 1
糖果右大于左 2 2 2 1
假如从5开始看,5>4所以在4的1的基础上加1,变成2…..这么看就发现不对了,所以是从右边往前
ratings数组:1 2 2 5 4 3 2
糖果右大于左1 2 1 2 1 1 1
糖果右大于左1 1 1 4 3 2 1
现在有两组,要让他同时满足两边,所以是取最大值 candyVec[i]只有取最大的才能既保持对左边candyVec[i - 1]的糖果多,也比右边candyVec[i + 1]的糖果多 。
ratings数组:1 2 2 5 4 3 2
糖果满足: 1 2 1 4 3 2 1
最后遍历这个数组输出结果即可
有两边的情况通常情况下要分开考虑
代码
class Solution {
public int candy(int[] ratings) {
int[] candy=new int[ratings.length];
candy[0]=1;
//从左往右遍历,右边孩子大于左边的情况
for(int i=1;i<ratings.length;i++){
if(ratings[i]>ratings[i-1]){
candy[i]=candy[i-1]+1;
}else{
candy[i]=1;
}
}
//从右往左遍历,左边孩子大于右边的情况
for(int i=ratings.length-2;i>=0;i--){
if(ratings[i]>ratings[i+1]){
candy[i]=Math.max(candy[i+1]+1,candy[i]);
}
}
int result=0;
for(int c:candy){
result+=c;
}
return result;
}
}