leetcode 135. 分发糖果 hard
题目描述:
老师想给孩子们分发糖果,有 N 个孩子站成了一条直线,老师会根据每个孩子的表现,预先给他们评分。
你需要按照以下要求,帮助老师给这些孩子分发糖果:
每个孩子至少分配到 1 个糖果。
相邻的孩子中,评分高的孩子必须获得更多的糖果。
那么这样下来,老师至少需要准备多少颗糖果呢?
示例 1:
输入: [1,0,2]
输出: 5
解释: 你可以分别给这三个孩子分发 2、1、2 颗糖果。
解题思路:
贪心,O(n)
首先给每个孩子分配一个保底的糖果,然后
- 从左到右遍历数组,保证每个位置和它的左邻居都符合逻辑,如果当前位置的评分比左边的高就把当前孩子的糖果数设为左边的糖果数+1
- 从右到左遍历数组,保证每个位置和它的右邻居都符合逻辑,这一次假如当前位置的评分比右边的高就把当前孩子的糖果数设为原来的糖果数和右边的糖果数+1 这两者之间的最大值,这样一来就保证了第二次遍历数组,每个位置的糖果数只增不减,所以不会破坏第一遍遍历的逻辑
代码:
class Solution {
public:
int candy(vector<int>& ratings) {
//贪心
if(ratings.size()<2)
return ratings.size();
vector<int> vec(ratings.size(),1);
//从左到右遍历数组,保证每个位置和它的左邻居都符合逻辑
for(int i=1;i<vec.size();++i){
if(ratings[i]>ratings[i-1])
vec[i]=vec[i-1]+1;
}
//从右到左遍历数组,保证每个位置和它的右邻居都符合逻辑
for(int i=vec.size()-2;i>=0;--i){
if(ratings[i]>ratings[i+1])
vec[i]=max(vec[i],vec[i+1]+1); //因为第二次遍历数组,每个位置的糖果数只增不减,所以不会破坏之前的逻辑
}
int res=0;
for(auto i:vec)
res +=i;
return res;
}
};