135-分发糖果

  1. 题目
    给出一组孩子评分,要求;
    a 每个孩子至少获得一颗糖果
    b 相邻两个孩子,评分高的孩子糖果数多
    问题:最少共需要多少颗糖果

  2. 思路[谷歌高畅力扣刷题笔记]
    从前往后遍历,如果后一个孩子评分高,保证后一个孩子糖果数多
    从后往前遍历,如果前一个孩子评分高,保证前一个孩子糖果数多(这里要保证前一个孩子和其左右两个孩子的比较,故使用max)

  3. 代码

# include
# include     // accumulate头函数
int candy(vector<int>& ratings) {
    int n = ratings.size();
    if(n<2) {   // 0/1个孩子的情况
        return n;
    }
    vector<int> nums(n, 1);     // 每个孩子的糖果数,至少获得一个糖果
    for(int i=1; i<n; i++) {    // 从前往后遍历;后一个孩子评分高,后一个孩子糖果数多
        if(ratings[i] > ratings[i-1]) {
            nums[i] = nums[i-1] + 1;
        }
    }
    for(int i=n-1; i>0; i--) {  // 从后往前遍历;前一个孩子评分高,使用max确保前一个孩子糖果数多
        if(ratings[i] < ratings[i-1]) {
            nums[i-1] = max(nums[i-1], nums[i]+1);
        }
    }
    return accumulate(nums.begin(), nums.end(), 0);     // 计算所有孩子的糖果总数;accumulate设置初值为0
}

你可能感兴趣的:(leetcode题目整理,算法,数据结构,leetcode)