有 N 个小孩站成一列。每个小孩有一个评级。
按照以下要求,给小孩分糖果:
-
每个小孩至少得到一颗糖果。
-
评级越高的小孩可以得到更多的糖果。
需最少准备多少糖果?
给定评级 = [1, 2]
, 返回 3
.
给定评级 = [1, 1, 1]
, 返回 3
.
给定评级 = [1, 2, 2]
, 返回 4
. ([1,2,1]).
/*
,一开始可以假设每个小孩都1颗糖果,如果第i个小孩比i-1个小孩rating高,那么糖果数也得比他高,和i+1个小孩比也是这样,于是可以从前往后扫描一遍,再从后往前扫描一遍。
先来看看两遍遍历的解法,首先初始化每个人一个糖果,然后这个算法需要遍历两遍,第一遍从左向右遍历,如果右边的小盆友的等级高,等加一个糖果,这样保证了一个方向上高等级的糖果多。然后再从右向左遍历一遍,如果相邻两个左边的等级高,而左边的糖果又少的话,则左边糖果数为右边糖果数加一。最后再把所有小盆友的糖果数都加起来返回即可。
*/
public class Solution {
public int candy(int[] ratings) {
// Write your code here
int len = ratings.length;
int[] dp = new int[len];
//1.先初始化每人一个糖果
for(int i = 0;i<len;i++){
dp[i] = 1;
}
//2.从左到右遍历判断
for(int i = 0;i<len - 1;i++){
if(ratings[i] < ratings[i+1]){
dp[i+1] = Math.max(dp[i+1], dp[i] + 1);
}
}
//3.从右到左
for(int i = len - 1;i>0;i--){
if(ratings[i] < ratings[i-1]){
dp[i-1] = Math.max(dp[i-1], dp[i] + 1);
}
}
int result = 0;
for(int i = 0;i<len;i++){
result += dp[i];
}
return result;
}
}