解法
- 因为
每个孩子至少分配到 1 个糖果。
所有孩子的糖果数量初始化为1 - 从左向右遍历,当左孩子比右孩子评分小,右孩糖果数量更新为
左孩数量+1
- 从右向左遍历,当左孩子比右孩子评分大,并且左孩糖果数量不大于右孩,则左孩糖果数量更新为
右孩数量+1
- 求和
举例:[13221]
:
- [1,1,1,1,1]
- [1,2,1,1,1]
- [1,2,1,2,1]
- 求和为7
class Solution {
public int candy(int[] ratings) {
int size = ratings.length;
if (size <= 1){
return size;
}
int[] candy = new int[size];
//初始化
for (int i = 0; i < size; i++) {
candy[i] = 1;
}
//左到右
for (int i = 0; i < size - 1; i++) {
if (ratings[i] < ratings[i + 1]){
candy[i + 1] = candy[i] + 1;
}
}
//右到左
for (int i = size - 1; i > 0; i--) {
if (ratings[i - 1] > ratings[i]){
candy[i - 1] = Math.max(candy[i] + 1, candy[i - 1]);
}
}
return Arrays.stream(candy).sum();
}
}
下面这段代码只是将初始化的for循环与左到右遍历结合,并且在右到左遍历中求和。
class Solution {
public int candy(int[] ratings) {
int size = ratings.length;
if (size <= 1){
return size;
}
int[] candy = new int[size];
int count = 0;//记录总糖果数
//初始化
candy[0] = 1;
//左到右
for (int i = 0; i < size - 1; i++) {
if (ratings[i] < ratings[i + 1]){
candy[i + 1] = candy[i] + 1;
continue;
}
candy[i + 1] = 1;
}
//右到左
for (int i = size - 1; i > 0; i--) {
if (ratings[i - 1] > ratings[i]){
candy[i - 1] = Math.max(candy[i] + 1, candy[i - 1]);
}
count += candy[i];
}
return count + candy[0];
}
}
原题 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 ∗ 1 0 4 1 <= n <= 2 * 10^4 1<=n<=2∗104
-
0 < = r a t i n g s [ i ] < = 2 ∗ 1 0 4 0 <= ratings[i] <= 2 * 10^4 0<=ratings[i]<=2∗104