题目说相邻的两个孩子中评分更高的孩子获得的糖果更多,表示我们既要考虑到跟左边的孩子比较,也要考虑右边的孩子,但是我们如果两边一起考虑一定会顾此失彼。这里就引入一个思想:先满足右边大于左边时的糖果分发情况,再满足左边大于右边时的糖果分发情况。需要注意的是,在满足后者的情况时,我们用到贪心的思想:取candyVec[i + 1] + 1 (满足后者情况时的糖果数量)和 candyVec[i] (满足前者情况时得到的糖果数量)中较大的糖果数量,保证第i个小孩的糖果数量既大于左边的也大于右边的。大家可以结合下面的代码及注释理解此题。
代码及注释如下:
class Solution {
public:
int candy(vector<int>& ratings) {
vector<int> candy(ratings.size(),1);
//先满足右边大于左边时的糖果分发情况
for(int i = 1;i < ratings.size();i++){
if(ratings[i] > ratings[i - 1]){
candy[i] = candy[i - 1] + 1;
}
}
//再满足左边大于右边时的糖果分发情况
for(int i = ratings.size() - 2;i >= 0;i--){
if(ratings[i] > ratings[i + 1]){
//取candyVec[i + 1] + 1 和 candyVec[i] 最大的糖果数量,
//保证第i个小孩的糖果数量既大于左边的也大于右边的
candy[i] = max(candy[i],candy[i + 1] + 1);
}
}
int sum = 0;
for(int i = 0;i < candy.size();i++){
sum += candy[i];
}
return sum;
}
};