题解 | #分糖果问题#

class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * pick candy
     * @param arr int整型vector the array
     * @return int整型
     */
    int candy(vector<int>& arr) {

        vector<int> nums(arr.size(),1);

        for(int i = 1 ;i  < arr.size(); i++) if (arr[i] > arr[i-1]) nums[i] = nums[i-1] + 1;

        for(int i = arr.size() - 2; i >=0 ; i--) if(arr[i] > arr[i+1]) nums[i] = nums[i] < nums[i+1] + 1 ? nums[i + 1] + 1 : nums[i];

        int sum = 0;
        
        for (const auto& val : nums) sum += val;
            
        
        return sum;

    }
};