老师想给孩子们分发糖果,有 N 个孩子站成了一条直线,老师会根据每个孩子的表现,预先给他们评分。
你需要按照以下要求,帮助老师给这些孩子分发糖果:
每个孩子至少分配到 1 个糖果。
相邻的孩子中,评分高的孩子必须获得更多的糖果。
那么这样下来,老师至少需要准备多少颗糖果呢?
示例 1:
输入: [1,0,2]
输出: 5
解释: 你可以分别给这三个孩子分发 2、1、2 颗糖果。
示例 2:
输入: [1,2,2]
输出: 4
解释: 你可以分别给这三个孩子分发 1、2、1 颗糖果。
第三个孩子只得到 1 颗糖果,这已满足上述两个条件。
思路:
在处理这道题的时候,不能只单纯的判断数字是否有递增;
在评分处于递增,但中间有递减,再次递增的情况时,要注意再次递减时候糖果的分配
评分: 1 2 3 3 4 5 4 3 4 5 6 7 6 5
糖果: 1 1 1 1 1 1 1 1 1 1 1 1 1 1 //每个小孩最少要拿到一个糖果
过程: 1 2 3 1 2 3 1 1 2 3 4 5 1 1 //后一个小朋友大于前一个时,要多拿一个糖果
———— ————
//5 4 //6 5 糖果分配不合理
//所以要从后向前再进行一次遍历,保证处于递减情况时,糖果分配的合理性
//糖果分配的第一个过程:if(ratings[i] > ratings[i-1]) 糖果比前一个多拿一个
//从后向前再次进行遍历时: nums[i] = max(nums[i],nums[i+1]+1);
//i的评分高于i+1,所以糖果一定要取最多的
//方法1:
class Solution {
public:
int candy(vector<int>& ratings) {
int len = ratings.size();
if(len == 0 || len == 1) return len;
int candy = 0;
vector<int> nums(len,1);
for(int i = 1;i < len;++i)
{
if(ratings[i] > ratings[i-1]) //比前一个评分高的情况
{
nums[i] = nums[i-1] + 1;
}
}
for(int i = len-2;i >= 0;--i) //比后一个评分高,但是分的糖果少的情况
{
if(ratings[i] > ratings[i+1])
{
nums[i] = max(nums[i],nums[i+1]+1);
}
}
for(int i = 0;i < nums.size();++i)
{
candy += nums[i];
}
return candy;
}
};
//方法2:
class Solution {
public:
int candy(vector<int>& ratings) {
int len = ratings.size();
if(len == 0 || len == 1) return len;
vector<int> nums(len,1); //存放每个人的糖果数
int candys = len;
for(int i = 1;i < len;++i) //正序
{
if(ratings[i] > ratings[i-1] && nums[i] <= nums[i-1])
{
candys += nums[i-1] - nums[i] + 1;
nums[i] = nums[i-1] + 1;
}
}
//并不是严格意义上的最少糖果
for(int i = len-1;i > 0;--i) //逆序
{
if(ratings[i] < ratings[i-1] && nums[i] >= nums[i-1])
{
candys += nums[i] - nums[i-1] + 1;
nums[i-1] = nums[i]+1;
}
}
//逆序进行遍历,确认是最少糖果
return candys;
}
};
博客参考(๑•̀ㅂ•́)و✧:https://blog.csdn.net/sinat_15723179/article/details/81410012