题解 | 分糖果问题

/**
 * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
 *
 * pick candy
 * @param arr int整型一维数组 the array
 * @return int整型
 */
function candy(arr) {
    // 一开始初始化为1
    if (arr.length === 1) {
        return 1;
    }
    let result = new Array(arr.length).fill(1);
    // 从左往右比较
    for (let i = 1; i < arr.length; i++) {
        if (arr[i] > arr[i - 1]) {
            result[i] = result[i - 1] + 1;
        }
    }
    // 从右往左比较
    for (let i = arr.length - 1; i > 0; i--) {
        if (arr[i] < arr[i - 1]) {
            result[i - 1] = Math.max(result[i-1],result[i] + 1);
        }
    }
    return result.reduce((pre, acc) => pre + acc, 0);
}
module.exports = {
    candy: candy,
};