这一篇也是基于"打家劫舍"的扩展,需要针对特殊情况特殊考虑,当然其本质还是动态规划,优化时需要考虑数据结构。

原题

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

计算在不触动警报的情况下,小偷一晚能够盗取的最高金额。

示例 1:

输入: [3,2,3,null,3,null,1]

     3
/ \
2 3
\ \
3 1 输出: 7
解释: 小偷一晚能够盗取的最高金额 = 3 + 3 + 1 = 7.

示例 2:

输入: [3,4,5,1,3,null,1]

     3
/ \
4 5
/ \ \
1 3 1 输出: 9
解释: 小偷一晚能够盗取的最高金额 = 4 + 5 = 9.

原题url:https://leetcode-cn.com/problems/house-robber-iii/

解题

先给出树节点的结构:

public class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) { val = x; }
}

简单思路

这道题简单来说,就是如果存在父节点、子节点、孙子节点三层的话,要么偷父节点 + 孙子节点,要么只偷子节点

顺着这个思路,我们只要找出每个节点所能偷到的最大值,自然也就能找出从 root 节点开始偷的最大值了。

接下来我们看看代码:

class Solution {

    Map<TreeNode, Integer> cache = new HashMap<>();

    public int rob(TreeNode root) {
if (root == null) {
return 0;
}
// 是否已经计算过
if (cache.containsKey(root)) {
return cache.get(root);
} // 策略1:抢当前节点和孙子节点
int sum1 = root.val +
// 左子节点的子节点们
(root.left == null ? 0 : (rob(root.left.left) + rob(root.left.right))) +
// 右子节点的子节点们
(root.right == null ? 0 : (rob(root.right.left) + rob(root.right.right)));
// 策略2:只抢子节点
int sum2 = rob(root.left) + rob(root.right);
// 找出更大的值
int sum = Math.max(sum1, sum2);
// 并记录
cache.put(root, sum);
return sum;
}
}

提交OK,执行用时:5 ms,只战胜了52.00%的 java 提交记录,因此还是有值得优化的地方。

优化

上面的解法,如果说有什么值得优化的地方,就是在于我们在动态规划时,不仅考虑了子节点,甚至也考虑到了孙子节点,因此当 子节点 变成 父节点 之后,孙子节点 也变成了 子节点。

也就是说,一开始的孙子节点被计算了两遍。虽然我们借用了一个 map 来记录了中间结果,但我们需要注意,这种情况依旧会被计算,只是代价被转移到了针对 map 的操作,这也是需要消耗时间的。

那么现在的优化,就转变成针对中间状态的记录上了。

其实我们针对每个节点的状态,只需要记录两种情况:抢或者不抢。而且这个状态只会被父节点用到,并不需要永久保留。因此我们考虑用一个长度为 2 的数组进行记录,这样就会快捷很多。

接下来我们看看代码:

class Solution {
public int rob(TreeNode root) {
// index为0,代表不抢当前节点的最大值
// index为1,代表抢当前节点,不抢子节点的最大值
int[] res = dp(root);
return Math.max(res[0], res[1]); } public int[] dp(TreeNode cur) {
if(cur == null) {
return new int[]{0,0};
} int[] left = dp(cur.left);
int[] right = dp(cur.right);
// 抢当前节点,子节点都不抢
int rob = cur.val + left[0] +right[0];
// 不抢当前节点,获取左右子节点各自的最大值
int notRob = Math.max(left[0], left[1]) + Math.max(right[0], right[1]);
// 返回结果
return new int[]{notRob, rob}; }
}

提交OK,时间消耗只有1 ms,确实快了很多。

总结

以上就是这道题目我的解答过程了,不知道大家是否理解了。这道题主要还是利用动态规划,只是需要大家进行思路转化,优化时需要考虑的更多是对数据结构的理解。

有兴趣的话可以访问我的博客或者关注我的公众号、头条号,说不定会有意外的惊喜。

https://death00.github.io/

公众号:健程之道

力扣337——打家劫舍 III的更多相关文章

  1. 刷题-力扣-337. 打家劫舍 III

    337. 打家劫舍 III 题目链接 来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/house-robber-iii 著作权归领扣网络所有.商 ...

  2. Java实现 LeetCode 337 打家劫舍 III(三)

    337. 打家劫舍 III 在上次打劫完一条街道之后和一圈房屋后,小偷又发现了一个新的可行窃的地区.这个地区只有一个入口,我们称之为"根". 除了"根"之外,每 ...

  3. 【力扣】337. 打家劫舍 III

    在上次打劫完一条街道之后和一圈房屋后,小偷又发现了一个新的可行窃的地区.这个地区只有一个入口,我们称之为"根". 除了"根"之外,每栋房子有且只有一个" ...

  4. 刷题-力扣-213. 打家劫舍 II

    213. 打家劫舍 II 题目链接 来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/house-robber-ii/ 著作权归领扣网络所有.商业 ...

  5. [LeetCode] 337. 打家劫舍 III (树形dp)

    题目 在上次打劫完一条街道之后和一圈房屋后,小偷又发现了一个新的可行窃的地区.这个地区只有一个入口,我们称之为"根". 除了"根"之外,每栋房子有且只有一个&q ...

  6. Leetcode 337. 打家劫舍 III

    题目链接 https://leetcode.com/problems/house-robber-iii/description/ 题目描述 在上次打劫完一条街道之后和一圈房屋后,小偷又发现了一个新的可 ...

  7. leetcode 337. 打家劫舍iii

    题目描述: 在上次打劫完一条街道之后和一圈房屋后,小偷又发现了一个新的可行窃的地区.这个地区只有一个入口,我们称之为“根”. 除了“根”之外,每栋房子有且只有一个“父“房子与之相连.一番侦察之后,聪明 ...

  8. LeetCode 337. 打家劫舍 III(House Robber III)

    题目描述 小偷又发现一个新的可行窃的地点. 这个地区只有一个入口,称为“根”. 除了根部之外,每栋房子有且只有一个父房子. 一番侦察之后,聪明的小偷意识到“这个地方的所有房屋形成了一棵二叉树”. 如果 ...

  9. 337. 打家劫舍 III(树上dp)

    在上次打劫完一条街道之后和一圈房屋后,小偷又发现了一个新的可行窃的地区.这个地区只有一个入口,我们称之为"根". 除了"根"之外,每栋房子有且只有一个" ...

随机推荐

  1. 关于better-scroll的使用注意事项

    better-scroll的原理 1.什么是 better-scroll? better-scroll 是一个移动端滚动的解决方案,它不仅可以做普通的滚动列表,还可以做轮播图.picker 等等. 2 ...

  2. 使用 node.js三行代码实现手机端访问html页面文件

    首先确保你安装了node (全局安装) npm install -g browser-sync // --files 路径是相对于运行该命令的项目(目录) browser-sync start --s ...

  3. 2018.12.7 浪在ACM 集训队第八次测试赛

    2018.12.7 浪在ACM 集训队第八次测试赛  https://blog.csdn.net/QLU_minoz/article/details/84886717   感谢苗学林同学C题和D题题解 ...

  4. Java面向对象程序设计第14章3-8和第15章6

    Java面向对象程序设计第14章3-8和第15章6 3.完成下面方法中的代码,要求建立一个缓冲区,将字节输入流中的内容转为字符串. import java.io.*; public class tes ...

  5. VS Code 解决 因为在此系统上禁止运行脚本

    vscode执行命令的 主要是由于没有权限执行脚本.开通权限就可以解决啦 在搜索框中输入:powerShell 选择管理员身份运行 输入命令行:set-ExecutionPolicy RemoteSi ...

  6. 洛谷P4141 消失之物 题解 背包问题扩展

    题目链接:https://www.luogu.com.cn/problem/P4141 题目大意: 有 \(n\) 件物品,求第 \(i\) 件物品不能选的时候(\(i\) 从 \(1\) 到 \(n ...

  7. 「2018-12-02模拟赛」T1 最短路 解题报告

    1.最短路(short.pas/cpp/in/out) 问题描述: 小 C 终于被小 X 感动了,于是决定与他看电影,然而小 X 距离电影院非常远,现在假设 每条道路需要花费小 X 的时间为 1,由于 ...

  8. 1050 螺旋矩阵 (25 分)C语言

    本题要求将给定的 N 个正整数按非递增的顺序,填入"螺旋矩阵".所谓"螺旋矩阵",是指从左上角第 1 个格子开始,按顺时针螺旋方向填充.要求矩阵的规模为 m 行 ...

  9. webpack 实时编译typescript与scss

    webpack.config.js const path = require('path'); const CopyWebpackPlugin = require('copy-webpack-plug ...

  10. C++中重载、重写(覆盖)和隐藏的区别

    转载自:https://blog.csdn.net/zx3517288/article/details/48976097 基本概念: 重载:是指同一可访问区内被声明的几个具有不同参数列(参数的类型,个 ...