-
题目
给出一组孩子评分,要求;
a 每个孩子至少获得一颗糖果
b 相邻两个孩子,评分高的孩子糖果数多
问题:最少共需要多少颗糖果 -
思路[谷歌高畅力扣刷题笔记]
从前往后遍历,如果后一个孩子评分高,保证后一个孩子糖果数多
从后往前遍历,如果前一个孩子评分高,保证前一个孩子糖果数多(这里要保证前一个孩子和其左右两个孩子的比较,故使用max) -
代码
# include<vector>
# include <numeric> // accumulate头函数
int candy(vector<int>& ratings) {
int n = ratings.size();
if(n<2) { // 0/1个孩子的情况
return n;
}
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 i=n-1; i>0; i--) { // 从后往前遍历;前一个孩子评分高,使用max确保前一个孩子糖果数多
if(ratings[i] < ratings[i-1]) {
nums[i-1] = max(nums[i-1], nums[i]+1);
}
}
return accumulate(nums.begin(), nums.end(), 0); // 计算所有孩子的糖果总数;accumulate设置初值为0
}