力扣解题思路:打家劫舍

213. 打家劫舍 II

思路:题目:
你是一个专业的小偷,计划偷窃沿街的房屋,每间房内都藏有一定的现金。这个地方所有的房屋都围成一圈,这意味着第一个房屋和最后一个房屋是紧挨着的。同时,相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。给定一个代表每个房屋存放金额的非负整数数组,计算你在不触动警报装置的情况下,能够偷窃到的最高金额。
这一题与198. 打家劫舍 唯一的区别是此时nums可以视为一个环形数组,回忆一下198题,由于不可以在相邻的房屋闯入,所以在当前位置 n 房屋可盗窃的最大值,要么就是 n-1 房屋可盗窃的最大值,要么就是 n-2 房屋可盗窃的最大值加上当前房屋的值,二者之间取最大值 因此可以使用动态规划算法:
动态规划方程:dp[n] = Math.max( dp[n-1], dp[n-2] + num ),198题的完整代码如下:

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

但是打家劫舍 II显然没有这么简单,因为该数组可以看作一个环形,如果偷了第一家,那么最后一家就不能偷,如果开始偷的是第二家,那么才可以偷最后一家,那么我们完全可以将其分成两种情况分别处理:

public int rob(int[] nums) {
 if(nums==null||nums.length==0)
  return 0;
 if(nums.length==1)
  return nums[0];
 int[] dp1 = new int[nums.length+1];
 int[] dp2 = new int[nums.length+1];
 dp1[1] = nums[0]; //从第1个房屋开始偷
 dp2[1] = 0; //从第2个房屋开始偷
 for(int i=2;i<nums.length;i++) {
  dp1[i] = Math.max( dp1[i-2]+nums[i-1],dp1[i-1] );
 }
    for(int i=2;i<nums.length+1;i++){
        dp2[i] = Math.max( dp2[i-2]+nums[i-1], dp2[i-1]);
    }
 return Math.max( dp1[nums.length-1], dp2[nums.length] );
}

当然,把这个动态规划封装成一个函数myRob也可以,封装好了就可以直接调用了,例如:

return Math.max(myRob(Arrays.copyOfRange(nums, 0, nums.length - 1))myRob(Arrays.copyOfRange(nums, 1, nums.length))

337. 打家劫舍 III


思路:在上次打劫完一条街道之后和一圈房屋后,小偷又发现了一个新的可行窃的地区。这个地区只有一个入口,我们称之为“根”。 除了“根”之外,每栋房子有且只有一个“父“房子与之相连。一番侦察之后,聪明的小偷意识到“这个地方的所有房屋的排列类似于一棵二叉树”。 如果两个直接相连的房子在同一天晚上被打劫,房屋将自动报警。计算在不触动警报的情况下,小偷一晚能够盗取的最高金额。

这一题与上面几题的区别是这一题考察到了树,那么对于任意一个节点,我们可以选择抢或者不抢该节点,然后我们对每个节点进行递归,计算抢该节点和不抢该节点所能得到的最大金额,将其保存在一个二元数组中。

但是我最开始想的是,一个节点偷与不偷是与其父节点有关的,那么我们遍历的时候是否需要记录其父节点的二元数组呢?其实是没必要的,因为我们采用递归的方式,实际上只有递归到叶子节点数组才会有值,然后再从叶子节点更新到根节点。因此,无需记录父节点数组,只需记录其子节点数组即可,将其作为返回值不就行了吗。
则有:

    result[0] = Math.max(left[0], left[1]) + Math.max(right[0], right[1]);    //计算不抢劫当前根节点可获得的最大金额(那么其左右子树可以随便抢)
    result[1] = left[0] + right[0] + root.val;//计算若抢劫根节点可获得的最大金额(此时,其左右子树的根节点不能被抢)

这样最终计算出的二元数组取其中一个最大值就是我们的答案了,完整代码如下:

public int rob(TreeNode root) {
    int[] result = robInternal(root);
    return Math.max(result[0], result[1]);
}
public int[] robInternal(TreeNode root) {   
    if (root == null) return new int[2];   
    int[] result = new int[2];
    int[] left = robInternal(root.left);    
    int[] right = robInternal(root.right);
    result[0] = Math.max(left[0], left[1]) + Math.max(right[0], right[1]);    
    result[1] = left[0] + right[0] + root.val;
    return result;
}

当然也可以不用数组来存放结果,这样就不用另写递归函数了:

public int rob(TreeNode root) {
    if (root == null) return 0;
    int val1 = root.val;
    if (root.left != null) val1 += rob(root.left.left) + rob(root.left.right);
    if (root.right != null) val1 += rob(root.right.left) + rob(root.right.right);
    int val2 = rob(root.left) + rob(root.right);
    return Math.max(val1, val2);
}

是不是更简单呢(✿◕‿◕✿)

  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值