一、题目
n 个孩子站成一排。给你一个整数数组 ratings 表示每个孩子的评分。
你需要按照以下要求,给这些孩子分发糖果:
每个孩子至少分配到 1 个糖果。
相邻两个孩子评分更高的孩子会获得更多的糖果。
请你给每个孩子分发糖果,计算并返回需要准备的 最少糖果数目 。
示例 1:
输入:ratings = [1,0,2]
输出:5
解释:你可以分别给第一个、第二个、第三个孩子分发 2、1、2 颗糖果。
示例 2:
输入:ratings = [1,2,2]
输出:4
解释:你可以分别给第一个、第二个、第三个孩子分发 1、2、1 颗糖果。
第三个孩子只得到 1 颗糖果,这满足题面中的两个条件。
提示:
n == ratings.length
1 <= n <= 2 * 104
0 <= ratings[i] <= 2 * 104
二、题解
这里与455一样,存在着比较的贪心策略,但却不需要进行排序和选择,因为这里题目要求与相邻的元素相比,所以我们可以依靠两次遍历来进行操作。
首先,因为要求每个孩子必须有一个糖果,所以我们定义了一个值为一的糖果分配数组。
然后第一次遍历:从前往后遍历,如果该位置对应分数比前一个位置对应分数大,那么该位置糖果分配数组值为前一个位置糖果分配数组值加一。第二次遍历:从后往前遍历,如果该位置对应分数比后一个位置对应分数大,那么该位置糖果分配数组值本来也应该为后一个位置糖果分配数组值加一,但是可能在第一次遍历结束后已经出现了该位置糖果分配值大于前一个位置糖果分配值的情况。这时符合题目要求,不进行操作。所以应该比较这两种情况,取最大值。直到遍历完成。
最后,将糖果分配数组的值进行求和,得到最少需要分配多少糖果的总数。
三、代码
class Solution {
public:
int candy(vector<int>& ratings) {
int n=ratings.size();
vector<int> nums(n,1);
for(int i=1;i<n;i++){
if(ratings[i]>ratings[i-1]) nums[i]=nums[i-1]+1;
}
for(int j=n-1;j>0;j--){
if(ratings[j-1]>ratings[j]&&nums[j-1]<=nums[j])
nums[j-1]=max(nums[j]+1,nums[j-1]);
}
return accumulate(nums.begin(),nums.end(),0);
}
};