题解 | 打家劫舍(二)

/**
 * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
 *
 * 
 * @param nums int整型一维数组 
 * @return int整型
 */
function rob( nums ) {
    // write code here

    const rpb1 = (nums1) => {
        if (nums1.length === 0) return 0
        if (nums1.length === 1) return nums1[0]
        const arr = new Array(nums1.length)
        arr[0] = nums1[0]
        arr[1] = Math.max(nums1[0], nums1[1])

        for (let i = 2; i <nums1.length; i++) {
            arr[i] = Math.max(arr[i-2]+nums1[i], arr[i-1])
        }

        return arr[nums1.length-1]
    }

    return Math.max(rpb1(nums.slice(1)), rpb1(nums.slice(0,-1)))
}
module.exports = {
    rob : rob
};