题目 :135. 分发糖果(力扣)
n 个孩子站成一排。给你一个整数数组 ratings 表示每个孩子的评分。
你需要按照以下要求,给这些孩子分发糖果:
每个孩子至少分配到 1 个糖果。
相邻两个孩子评分更高的孩子会获得更多的糖果。
请你给每个孩子分发糖果,计算并返回需要准备的 最少糖果数目 。
示例 1:
输入:ratings = [1,0,2] 输出:5
解释:你可以分别给第一个、第二个、第三个孩子分发 2、1、2 颗糖果。
示例 2:
输入:ratings = [1,2,2] 输出:4
解释:你可以分别给第一个、第二个、第三个孩子分发 1、2、1 颗糖果。
第三个孩子只得到 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;
}
}