135. 分发糖果( 😦 😦 😦 😦 😦 😦 😦 😦)
老师想给孩子们分发糖果,有 N 个孩子站成了一条直线,老师会根据每个孩子的表现,预先给他们评分。
你需要按照以下要求,帮助老师给这些孩子分发糖果:
每个孩子至少分配到 1 个糖果。
评分更高的孩子必须比他两侧的邻位孩子获得更多的糖果。
那么这样下来,老师至少需要准备多少颗糖果呢?
思路:
每个元素均和两边元素比较:分两次遍历,第一次遍历从前向后,实现左规则,第二次遍历从后向前,实现右规则
(规则的定义以当前项即第i项为基准)
左规则:r[i]>r[i-1],则nums[i]=nums[i-1]+1,否则置1
右规则:若r[i]>r[i+1],且nums[i]<=nums[i+1],则nums[i+1]=nums[i]+1,否则不变
class Solution {
public int candy(int[] ratings) {
//每个元素均和两边元素比较(以当前项即第i项为基准):分两次遍历,第一次遍历从前向后,实现左规则,第二次遍历从后向前,实现右规则
int[] nums=new int[ratings.length];
if(ratings.length==0)
return 0;
//从前向后遍历,实现左规则:r[i]>r[i-1],则nums[i]=nums[i-1]+1,否则置1
for(int i=0;i<ratings.length;i++) {
if(i>0 && ratings[i]>ratings[i-1])
nums[i]=nums[i-1]+1;
else
nums[i]=1;
}
//从后向前遍历,实现右规则:若r[i]>r[i+1],且r[i]比r[i+1]得的少或相等,则nums[i+1]=nums[i]+1,否则不变
for(int i=ratings.length-2;i>=0;i--) {
if(ratings[i]>ratings[i+1] && nums[i]<=nums[i+1])
nums[i]=nums[i+1]+1;
}
int count=0;
for(int i=0;i<nums.length;i++)
count+=nums[i];
return count;
}
}
注意:贪心算法每次只考虑单个个体,即只考虑一侧的大小关系进行更新。