分发糖果(力扣135)

qy发大财2025-02-23 13:30

题目说相邻的两个孩子中评分更高的孩子获得的糖果更多，表示我们既要考虑到跟左边的孩子比较，也要考虑右边的孩子，但是我们如果两边一起考虑一定会顾此失彼。这里就引入一个思想：先满足右边大于左边时的糖果分发情况，再满足左边大于右边时的糖果分发情况。需要注意的是，在满足后者的情况时，我们用到贪心的思想：取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;
    }
};