Candy
原题链接Candy
n个孩子站成一排,分糖果。ratings[i]表示第i个孩子的等级,要求
- 每个孩子至少有一个糖果
- 高等级的孩子分得的糖果要高于相邻的低等级的孩子分得的糖果
计算最少需要的糖果数
初始时每个孩子一个糖果可以满足,相邻孩子的糖果数量需要从两方面考虑
首先,保证高等级的孩子分得的糖果比左边相邻的低等级的孩子分得的糖果多(由于求最少糖果数,所以只多一个即可)
然后,保证高等级的孩子分得的糖果比右边相邻的低等级的孩子分得的糖果多(同理也只多一个)
这样就满足比左右两边都多的情况
代码如下
class Solution {
public:
int candy(vector<int>& ratings) {
if(ratings.empty()) return 0;
int n = ratings.size();
/* 每个孩子至少分得一个糖果 */
vector<int> candies(n, 1);
/* 首先保证比左边的低等级的孩子分得的糖果多1 */
for(int i = 1; i < n; ++i)
candies[i] = (ratings[i] > ratings[i - 1]) ? candies[i - 1] + 1 : candies[i];
/* 然后保证比右边的多,这里需要比较本身的糖果数和右边糖果数+1的大小,选择最大的 */
for(int i = n - 2; i >= 0; --i)
candies[i] = (ratings[i] > ratings[i + 1]) ? std::max(candies[i], candies[i + 1] + 1) : candies[i];
/* 求和 */
return std::accumulate(candies.begin(), candies.end(), 0);
}
};