【原】135. Candy (Hard)

135. Candy (Hard)

题目描述

一群孩子站成一排，每一个孩子有自己的评分。现在需要给这些孩子发糖果，规则是如果一
个孩子的评分比自己身旁的一个孩子要高，那么这个孩子就必须得到比身旁孩子更多的糖果；所
有孩子至少要有一个糖果。求解最少需要多少个糖果。
输入输出样例
输入是一个数组，表示孩子的评分。输出是最少糖果的数量。

Input: [1,0,2]
Output: 5

在这个样例中，最少的糖果分法是 [2,1,2]。

解题思路

我们只需要简单的两次遍历即可：把所有孩子的糖果数初始化为 1；
先从左往右遍历一遍，如果右边孩子的评分比左边的高，则右边孩子的糖果数更新为左边孩子的
糖果数加 1；再从右往左遍历一遍，如果左边孩子的评分比右边的高，且左边孩子当前的糖果数
不大于右边孩子的糖果数，则左边孩子的糖果数更新为右边孩子的糖果数加 1。通过这两次遍历，
分配的糖果就可以满足题目要求了。这里的贪心策略即为，在每次遍历中，只考虑并更新相邻一
侧的大小关系。
在样例中，我们初始化糖果分配为 [1,1,1]，第一次遍历更新后的结果为 [1,1,2]，第二次遍历
更新后的结果为 [2,1,2]。

代码

class Solution {
public:
    int candy(vector<int>& ratings) {
        int size = ratings.size();
        if (size==1)
            return 1;
        // 建立一个容器  先全部初始化为1  表示都给一个糖果
        vector<int> num(size,1);    
        // 从左向右遍历 如果右边的大于左边的  右边的就+1
        int i;
        // 从1开始遍历  避免数组溢出问题
        for(i=1;i<size;i++)
        {
            if(ratings[i]>ratings[i-1])
                num[i]=num[i-1]+1;
        }
        // 从右向左遍历  如果左边的大于右边的  左边的就+1
        for(i=size-1;i>0;i--)
        {
            if(ratings[i]<ratings[i-1])
                num[i-1]=max(num[i-1],num[i]+1);
        }
        return std::accumulate(num.begin(),num.end(),0);
    }
};