LeetCode 198. 打家劫舍 （动态规划 + 优化）—— JavaScript

题链接：https://leetcode-cn.com/problems/house-robber/

---

题描述：

---

解题思路：

设置一个 dp 数组，dp[i] 表示在当前位置下，不触发警报能偷窃到的最大金额。

则：dp[i] 等于与 i 不相邻的左边的 max(dp[0] — dp[i - 2]) 加上 nums[i]。

dp[0] 为 nums[0]，dp[1] 为 nums[1]。

最后返回 dp[] 中的最大值即可。根据以上，我们写出第一版代码：

---

代码实现（第一版）：

/**
 * @param {number[]} nums
 * @return {number}
 */
var rob = function(nums) {
    if (nums.length == 0) {
        return 0;
    }
    if (nums.length == 1) {
        return nums[0];
    }
    if (nums.length <= 2) {
        return nums[0] > nums[1] ? nums[0] : nums[1];
    }
    let dp = [];
    dp[0] = nums[0];
    dp[1] = nums[1];
    let res = dp[0] > dp[1] ? dp[0] : dp[1];
    for (let i = 2; i < nums.length; i++) {
        let t = 0;
        for (let j = 0; j <= i - 2; j++) {
            t = dp[j] > t ? dp[j] : t; 
        }
        dp[i] = t + nums[i];
        res = dp[i] > res? dp[i] : res;
    }
    return res;
};

时间复杂度：O(n * logn)

空间复杂度：O(n)

---

优化思路：

由于不能进入相邻的房间，所以 dp[i] 要么是 dp[i - 2] 加上 nums[i]，要么是 dp[i - 1]。

且 dp 数组并不是必须的，我们可以在原数组上进行计算。根据以上，我们写出优化后的代码：

---

代码实现（优化版）：

/**
 * @param {number[]} nums
 * @return {number}
 */
var rob = function(nums) {
    if (nums.length == 0) {
        return 0;
    }
    if (nums.length == 1) {
        return nums[0];
    }
    if (nums.length <= 2) {
        return nums[0] > nums[1] ? nums[0] : nums[1];
    }
    nums[1] = nums[0] > nums[1] ? nums[0] : nums[1];
    for (let i = 2; i < nums.length; i++) {
        nums[i] = Math.max(nums[i - 2] + nums[i], nums[i - 1]);
    }
    return nums[nums.length - 1];
};

时间复杂度：O(n)

空间复杂度：O(1)