题解 | #分糖果问题#

#include <vector>
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * pick candy
     * @param arr int整型vector the array
     * @return int整型
     */
    int candy(vector<int>& arr) {
        // write code here
        int n = arr.size();
        vector<int>candy(n,1);
        for(int i=1;i<n;i++) {
            if(arr[i]>arr[i-1]) candy[i]=candy[i-1]+1;//比左边的小孩多分发一个糖果
        }
        for(int i=n-2;i>=0;i--) {
            if(arr[i]>arr[i+1]) candy[i]=max(candy[i],candy[i+1]+1);//比右边的小孩多分发一个糖果
        }
        int ans=0;
        for(int i=0;i<n;i++) {
            ans+=candy[i];
        }
        return ans;
    }
};