/*本题给小孩分配糖果的规则:
1.每个小孩至少一个糖果。
2.高等级小孩的糖果比相邻的小孩的糖果多。
题目要求最小的糖果数目。直接的想法是:
1.先给每个小孩一个糖果。
2.从左往右扫描等级数组,再从右往左扫描一遍等级数组,保证第2个条件成立。
3.返回总糖果数目。
参考:https://github.com/soulmachine/leetcode*/
class Solution {
public:
int candy(vector<int> &ratings) {
vector<int> candies(ratings.size(), 1);//初始化,每个小孩有1个糖
for(int i = 1; i < ratings.size(); ++i){//从左往右扫描
if(ratings[i] > ratings[i-1]){
candies[i] = std::max(candies[i-1]+1, candies[i]);
}
}
for(int i = ratings.size()-2; i >= 0; --i){//从右往左扫描
if(ratings[i] > ratings[i+1]){
candies[i] = std::max(candies[i+1]+1, candies[i]);
}
}
return std::accumulate(candies.begin(), candies.end(), 0);
}
};
LeetCode之Candy
最新推荐文章于 2019-07-25 11:42:06 发布