代码随想录算法训练营Day48|LC198 打家劫舍&LC213 打家劫舍II&LC337 打家劫舍III

一句话总结：前两题白给，第三题树形DP有点难。

原题链接：198 打家劫舍

滚动数组直接秒了。

class Solution {
    public int rob(int[] nums) {
        int n = nums.length;
        int first = 0, second = nums[0];
        for (int i = 2; i <= n; ++i) {
            int tmp = Math.max(first + nums[i - 1], second);
            first = second;
            second = tmp;
        }
        return second;
    }
}

原题链接：213 打家劫舍II 

 

条件与上一题差不多，多了一个房屋成环的限制，那么只需要考虑两个可能性：

1. 偷第0家，那么剩下可偷的房屋就从第2到第n - 2家里面选择；
2. 不偷第0家，那么剩下的就从第1到第n - 1家里面选择。

剩下的房屋里的选择与上一题类似。最后得到两种可能性的最大值，返回即可。

class Solution {
    public int rob(int[] nums) {
        int n = nums.length;
        return Math.max(nums[0] + rob(nums, 2, n - 1), rob(nums, 1, n));
    }

    private int rob(int[] nums, int start, int end) {
        int first = 0, second = 0;
        for (int i = start; i < end; ++i) {
            int tmp = Math.max(second, first + nums[i]);
            first = second;
            second = tmp;
        }
        return second;
    }
}

原题链接：337 打家劫舍III

原分析在此处。这里涉及到递归和动态规划的结合。利用动规五部曲和递归三步逐步分析如下：

• 首先要考虑对树的遍历方式是前中后还是层序，这里需要后序遍历，往根节点的计算过程需要子节点的计算参与； 
• 然后确定递归函数的参数和返回值。对于某一具体节点房屋，只有选择偷和不偷两种状态，因此该递归函数的返回值为长度为2的数组；同时，参数即为当前节点。其实，这里返回的数组即为dp数组，由于递归的存在，递归过程中系统会将前面每一层的参数保存起来。
• 再就是确定递归的终止条件。在遍历遇到空节点时，无论偷不偷，当前节点返回的值都将会是0，因此这里的递归终止条件即为：

  if (cur == null) return new int[2];

  这里其实也就相当于对dp数组做了初始化。

• 紧接着需要确定的是动态规划过程中的遍历顺序。之前已经确定了要用后序遍历，因此遍历顺序为：通过递归遍历左子节点，获得偷或者不偷左子节点的金钱；通过递归遍历右子节点，获得偷或者不偷右子节点的金钱。

• 之后是确定单层递归的逻辑。如果选择偷当前节点，那么该节点的左右子节点就不能偷，于是有：val1 = cur.val + left[0] + right[0]; 如果选择不偷当前节点，那么该节点的左右子树就可以偷，具体偷不偷则要取决于最大化对该子节点的收益，即有：val2 = Math.max(left[0], left[1]) + Math.max(right[0], right[1])。这样即可得出当前节点的状态为：{val1, val2}。

以上即是分析过程，最后代码如下：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public int rob(TreeNode root) {
        int[] ans = robTree(root);
        return Math.max(ans[0], ans[1]);
    }

    private int[] robTree(TreeNode root) {
        if (root == null) return new int[2];
        int[] left = robTree(root.left);
        int[] right = robTree(root.right);
        int val1 = root.val + left[0] + right[0];
        int val2 = Math.max(left[0], left[1]) + Math.max(right[0], right[1]);
        return new int[]{val2, val1};
    }
}