老师想给孩子们分发糖果,有 N 个孩子站成了一条直线,老师会根据每个孩子的表现,预先给他们评分。
你需要按照以下要求,帮助老师给这些孩子分发糖果:
每个孩子至少分配到 1 个糖果。
相邻的孩子中,评分高的孩子必须获得更多的糖果。
那么这样下来,老师至少需要准备多少颗糖果呢?
leetcode
解题思路:贪心做法,通过两次遍历,求每个位置的元素满足右规则和左规则的最小情况,然后取最大值。
如果取一边情况下的最大值,会不会破坏另一边的规则呢?
再遍历每一个方向上的时候,我们都是在保证后者需要大于前者的时候,后者在前者的基础上加一,如果取两边最大,对于单独一边来说不会违反规则,比较一边的大小的时候,另一边的大小不会被破坏,但是如果不取最大,就不能满足一边的条件,也就无法求得最后的解。
class Solution {
public int candy(int[] ratings) {
int n = ratings.length;
int[] l = new int[n], r = new int[n];
// 初始化,保证每个人都有一个
Arrays.fill(r, 1);
Arrays.fill(l, 1);
for(int i = 1; i < n; i++){
// 如果左边节点小于当前节点,那么一定要大于左边节点值
if(ratings[i] > ratings[i - 1]) l[i] = l[i - 1] + 1;
}
for(int i = n - 2; i >= 0; i--){
// 如果右边节点小于当前节点,那么一定要大于右边节点值
if(ratings[i] > ratings[i + 1]) r[i] = r[i + 1] + 1;
}
int res = 0;
for(int i = 0; i < n; i++){
// 获取最大值
res += Math.max(r[i], l[i]);
}
return res;
}
}