代码随想录算法训练营第四十五天| 198.打家劫舍、 213.打家劫舍II、 337.打家劫舍III

LeetCode 198.打家劫舍

题目链接：https://leetcode.cn/problems/house-robber/description/
文章链接：https://programmercarl.com/0198.%E6%89%93%E5%AE%B6%E5%8A%AB%E8%88%8D.html

思路

 * dp[i]：考虑下标i（包括i）以内的房屋，最多可以偷窃的金额为dp[i]。
 * 那么对于第i个房屋偷与不偷取决于第i-1个房屋有没有偷
 * 若第i-1个房屋没有偷，则第i个房屋可以偷，则可以表示为dp[i-2] + nums[i]
 * 若考虑第i-1个房屋偷，则可以表示为dp[i-1]

    public int rob(int[] nums) {
        int[] dp = new int[nums.length];
        dp[0] = nums[0];
        dp[1] = Math.max(nums[0], nums[1]);
        for (int i = 2; i < nums.length; i++) {
            dp[i] = Math.max(dp[i - 2] + nums[i], dp[i - 1]);
        }
        return dp[nums.length-1];
    }

LeetCode 213.打家劫舍II

题目链接：https://leetcode.cn/problems/house-robber-ii/description/
文章链接：https://programmercarl.com/0213.%E6%89%93%E5%AE%B6%E5%8A%AB%E8%88%8DII.html

思路

这道题目和198.打家劫舍 (opens new window)是差不多的，唯一区别就是成环了。
主要分为两种情况

    public int rob(int[] nums) {
        int[] dp1 = new int[nums.length];
        int[] dp2 = new int[nums.length];
        dp1[0] = nums[0];
        dp1[1] = Math.max(nums[0], nums[1]);
        dp2[1] = nums[1];
        dp2[2] = Math.max(nums[1], nums[2]);
        for (int i = 2; i < nums.length - 1; i++) {
            dp1[i] = Math.max(dp1[i - 1], dp1[i - 2] + nums[i]);
        }
        for (int i = 3; i < nums.length; i++) {
            dp2[i] = Math.max(dp2[i - 1], dp2[i - 2] + nums[i]);
        }
        return Math.max(dp1[nums.length - 2], dp2[nums.length - 1]);
    }

LeetCode 337.打家劫舍III

题目链接：https://leetcode.cn/problems/house-robber-iii/description/
文章链接：https://programmercarl.com/0337.%E6%89%93%E5%AE%B6%E5%8A%AB%E8%88%8DIII.html#%E7%AE%97%E6%B3%95%E5%85%AC%E5%BC%80%E8%AF%BE

思路

 * 与198.打家劫舍，213.打家劫舍II一样，关键是要讨论当前节点抢还是不抢。
 * 如果抢了当前节点，两个孩子就不能动，如果没抢当前节点，就可以考虑抢左右孩子
 * 这里我们要求一个节点 偷与不偷的两个状态所得到的金钱，那么返回值就是一个长度为2的数组。
 * 所以dp数组（dp table）以及下标的含义：下标为0记录不偷该节点所得到的的最大金钱，下标为1记录偷该节点所得到的的最大金钱。
 * 如果是偷当前节点，那么左右孩子就不能偷，val1 = cur->val + left[0] + right[0]; （如果对下标含义不理解就再回顾一下dp数组的含义）
 * 如果不偷当前节点，那么左右孩子就可以偷，至于到底偷不偷一定是选一个最大的，所以：val2 = max(left[0], left[1]) + max(right[0], right[1]);
 * 最后当前节点的状态就是{val2, val1}; 即：{不偷当前节点得到的最大金钱，偷当前节点得到的最大金钱}

  public int rob(TreeNode root) {
        int[] dp = robAction1(root);
        return Math.max(dp[0], dp[1]);
    }
    private int[] robAction1(TreeNode root){
        int[] dp = new int[2];
        if (root == null)
            return dp;
        int[] left = robAction1(root.left);
        int[] right = robAction1(root.right);
        // 若不偷当前节点，则考虑偷左右孩子
        dp[0] = Math.max(left[0],left[1]) + Math.max(right[0],right[1]);
        // 若考虑偷当前节点
        dp[1] = root.val + left[0] + right[0];
        return dp;
    }