题目描述
一群孩子站成一排,每一个孩子有自己的评分。现在需要给这些孩子发糖果,规则是如果一 个孩子的评分比自己身旁的一个孩子要高,那么这个孩子就必须得到比身旁孩子更多的糖果;所有孩子至少要有一个糖果。求解最少需要多少个糖果。
解题思路
-
把所有孩子的糖果数初始化为 1; 先从左往右遍历一遍,如果右边孩子的评分比左边的高,则右边孩子的糖果数更新为左边孩子的 糖果数加1;
-
再从右往左遍历一遍,如果左边孩子的评分比右边的高,且左边孩子当前的糖果数 不大于右边孩子的糖果数,则左边孩子的糖果数更新为右边孩子的糖果数加1。
-
通过这两次遍历, 分配的糖果就可以满足题目要求了。这里的贪心策略即为,在每次遍历中,只考虑并更新相邻一 侧的大小关系。
AC
class Solution {
public:
int candy(vector<int>& ratings) {
int size = ratings.size();
if(size<2) return size;
vector<int> v(size,1);
//先从左到右遍历
for(int i=0;i<size-1;i++){
if(ratings[i+1]>ratings[i])
v[i+1]=v[i]+1;
}
//再从右到左遍历
for(int i=size-1;i>=1;i--){
if(ratings[i-1]>ratings[i])
v[i-1]=max(v[i-1],v[i]+1);
}
return accumulate(v.begin(),v.end(),0);
}
};
学如逆水行舟,不进则退