分糖果问题

描述

一群孩子做游戏，现在请你根据游戏得分来发糖果，要求如下：

1. 每个孩子不管得分多少，起码分到一个糖果。

2. 任意两个相邻的孩子之间，得分较多的孩子必须拿多一些糖果。(若相同则无此限制)

给定一个数组 arr 代表得分数组，请返回最少需要多少糖果。

要求: 时间复杂度为 O(n) 空间复杂度为 O(n)

数据范围： 1≤1000001≤n≤100000 ，1≤10001≤ai​≤1000

示例1

输入：

[1,1,2]

返回值：

4

说明：

最优分配方案为1,1,2 

示例2

输入：

[1,1,1]

返回值：

3

说明：

最优分配方案是1,1,1 

/**
 * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
 *
 * 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,
};