135.分发糖果
我的解法:
由于需要考虑相邻两个孩子的评分才能确定当前孩子获得的糖果数,因此需要正向和反向两次遍历评分数组分别考虑其左侧和右侧孩子评分的大小关系。
- 先定义一个大小为n,初值均为1的candy数组;
- 正向遍历ranting数组:左侧孩子评分更低时,当前孩子会比左孩子多获得一个糖果;
- 反向遍历rating数组:右侧孩子评分更低时,当前孩子获得的糖果数是正向遍历的糖果数与右孩子糖果数加1的最大值。
class Solution {
public:
int candy(vector<int>& ratings) {
int n = ratings.size(),total = 0;
vector<int> candy(n,1);
for(int i = 1; i < n; ++i){
if(ratings[i - 1] < ratings[i]){
candy[i] = candy[i - 1] + 1;
}
}
for(int i = n - 2; i >= 0; --i){
if(ratings[i] > ratings[i + 1]){
candy[i] = max(candy[i],candy[i + 1] + 1);
}
}
for(int x : candy){
total += x;
}
return total;
}
};