题目:给小孩分糖果,每个小孩至少要分一个糖果,如果这个小孩的rating比相邻的小孩高,那么他分的糖果也要比相邻小孩多,问最少需要多少糖果?
题目分析:使用额外空间candy[n]表示每个孩子的糖果数,在遍历过程中,如果第i+1位的孩子的ratings大于第i位孩子的ratings,那么给第i+1位的孩子糖果数=给第i位的孩子的糖果数+1。
如果孩子i+1的rate小于等于孩子i 的rate怎么办?我们不知道当前最好的选择是给孩子i+1多少糖果。解决方法是暂时不处理这种情况。等数组遍历完了,我们再一次从尾到头遍历数组,这回逆过来贪心,就可以处理之前略过的孩子。
最后累加candy[n]即得到最小糖果数。
import java.util.Arrays;
class Solution {
public static int candy(int[] ratings) {
int len = ratings.length;
int[] candy = new int[len];
Arrays.fill(candy, 1);
int candySum = 0;
for(int i = 1; i < len; i++){
if(ratings[i - 1] < ratings[i]){
candy[i] = candy[i - 1] + 1;
}
}
for(int i = len - 2; i >= 0; i--){
if(ratings[i] > ratings[i + 1]){
candy[i] = Math.max(candy[i + 1] + 1, candy[i]);
}
}
for(int i = 0; i < len; i++){
candySum += candy[i];
}
return candySum;
}
public static void main(String[] args) {
int[] nums = {12,4,3,11,34,34,1,67};
System.out.println(candy(nums));
}
}