Description:
有 N 个小孩站成一列。每个小孩有一个评级。
按照以下要求,给小孩分糖果:
-
每个小孩至少得到一颗糖果。
-
评级越高的小孩可以得到更多的糖果。
需最少准备多少糖果?
Explanation:
给定评级 = [1, 2]
, 返回 3
.
给定评级 = [1, 1, 1]
, 返回 3
.
给定评级 = [1, 2, 2]
, 返回 4
. ([1,2,1]).
Solution:
感觉题目描述不清楚,就查看了一些参考,题目描述应该为 相邻两个人,评级越高,获得的糖果越多
所以从前往后,再从后往前,遍历两次,比较相邻两点,并将糖果数记录在dp中。
public class Solution {
/**
* @param ratings Children's ratings
* @return the minimum candies you must give
*/
public int candy(int[] ratings) {
// Write your code here
int len = ratings.length;
int[] dp = new int[len];
for(int i = 0;i<len;i++){
dp[i] = 1;
}
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);
}
}
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;
}
}