0
点赞
收藏
分享

微信扫一扫

通过docker容器部署oceanbase单机版测试

一句话总结:前两题白给,第三题树形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};
    }
}
举报

相关推荐

0 条评论