打家劫舍

默认 leetcode

你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。

给定一个代表每个房屋存放金额的非负整数数组,计算你 不触动警报装置的情况下 ,一夜之内能够偷窃到的最高金额。

来源:力扣(LeetCode) 链接:https://leetcode.cn/problems/house-robber 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

这题一看就是动态规划,需要做的就是找出规律,有点类似于数学归纳。

简单验算一下

    // n = 1  i0
    // n = 2  max(i0, i1)
    // n = 3  max(i0 + i2, i1)
    // n = 4  max(i0 + i2, i1 + i3, i0 + i3) (n2 + i3, n1 + i2)
    // n = 5  max(n3 + i4, n2 + i3)
    // n = 6  max(n4 + i5, n3 + i4)

找到规律,代码就很简单了

var rob = function(nums) {
    const length = nums.length;
    const arr = Array(length).fill(0);
    arr[0] = nums[0];
    if (length > 1) {
        arr[1] = Math.max(nums[0], nums[1]);
    }
    if (length > 2) {
        arr[2] = Math.max(nums[0] + nums[2], nums[1]);
    }
    for (let i = 3; i < length; i++) {
        arr[i] = Math.max(arr[i - 3] + nums[i - 1], arr[i - 2] + nums[i]);
    }
    return arr[length - 1];
};

上一篇 下一篇