LeetCode:337. House Robber III(偷窃问题)

柔情只为你懂 2022-04-04 12:54 236阅读 0赞

文章最前: 我是Octopus,这个名字来源于我的中文名—章鱼;我热爱编程、热爱算法、热爱开源。所有源码在我的个人github ;这博客是记录我学习的点点滴滴,如果您对 Python、Java、AI、算法有兴趣,可以关注我的动态,一起学习,共同进步。

相关文章:

  1. LeetCode:55. Jump Game(跳远比赛)
  2. Leetcode:300. Longest Increasing Subsequence(最大增长序列)
  3. LeetCode:560. Subarray Sum Equals K(找出数组中连续子串和等于k

问目录:

题目描述:

java实现方法1:(递归算法)

python实现方式1:

java实现方法2:(另一种递归思路)

python实现方式2:

源码github地址:https://github.com/zhangyu345293721/leetcode


题目描述:

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

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

示例 1:

  1. 输入: [3,2,3,null,3,null,1]
  2. 3
  3. / \
  4. 2 3
  5. \ \
  6. 3 1
  7. 输出: 7
  8. 解释: 小偷一晚能够盗取的最高金额 = 3 + 3 + 1 = 7.

示例 2:

  1. 输入: [3,4,5,1,3,null,1]
  2. 3
  3. / \
  4. 4 5
  5. / \ \
  6. 1 3 1
  7. 输出: 9
  8. 解释: 小偷一晚能够盗取的最高金额 = 4 + 5 = 9.

来源:力扣(LeetCode)


java实现方法1:(递归算法)

  1. /**
  2. * 找出抢劫最大结果
  3. *
  4. * @param root 根节点
  5. * @return 最大值
  6. */
  7. public int rob(TreeNode root) {
  8. int[] result = helper(root);
  9. return Math.max(result[0], result[1]);
  10. }
  11. /**
  12. * 帮助方法
  13. *
  14. * @param node 节点
  15. * @return 数组
  16. */
  17. public int[] helper(TreeNode node) {
  18. if (node == null) {
  19. return new int[]{0, 0};
  20. }
  21. if (node.left == null && node.right == null) {
  22. return new int[]{node.val, 0};
  23. }
  24. int[] l = helper(node.left);
  25. int[] r = helper(node.right);
  26. return new int[]{node.val + l[1] + r[1], Math.max(l[0], l[1]) + Math.max(r[0], r[1])};
  27. }

时间复杂度:O(n)

空间复杂度:O(n)


python实现方式1:

  1. def helper(node: TreeNode) -> List[int]:
  2. '''
  3. 帮助方法
  4. Args:
  5. node: 节点
  6. Returns:
  7. 数组
  8. '''
  9. if node == None:
  10. return [0, 0]
  11. if not node.left and not node.right:
  12. return [node.val, 0]
  13. l = helper(node.left)
  14. r = helper(node.right)
  15. return [node.val + l[1] + r[1], max(r[0], r[1]) + max(l[0] + l[1])]
  16. def rob(root: TreeNode) -> int:
  17. '''
  18. 找出抢劫最大值
  19. Args:
  20. root: 跟节点
  21. Returns:
  22. 最大值
  23. '''
  24. result = helper(root)
  25. return max(result[0], result[1])

时间复杂度:O(n)

空间复杂度:O(n)


java实现方法2:(另一种递归思路)

  1. /**
  2. * 找出抢劫最大结果
  3. *
  4. * @param root 根节点
  5. * @return 最大值
  6. */
  7. public int rob2(TreeNode root) {
  8. if (root == null) {
  9. return 0;
  10. }
  11. int val = 0;
  12. if (root.left != null) {
  13. val += rob2(root.left.left) + rob2(root.left.right);
  14. }
  15. if (root.right != null) {
  16. val += rob2(root.right.left) + rob2(root.right.right);
  17. }
  18. return Math.max(val + root.val, rob2(root.left) + rob2(root.right));
  19. }

时间复杂度:O(n)

空间复杂度:O(n)


python实现方式2:

  1. def rob2(root: TreeNode) -> int:
  2. '''
  3. 找出抢劫最大值
  4. Args:
  5. root: 跟节点
  6. Returns:
  7. 最大值
  8. '''
  9. if not root:
  10. return 0
  11. val = 0
  12. if root.left :
  13. val += rob2(root.left.left) + rob2(root.left.right)
  14. if root.right :
  15. val += rob2(root.right.left) + rob2(root.right.right)
  16. return max(val + root.val, rob2(root.right) + rob2(root.left))

时间复杂度:O(n)

空间复杂度:O(n)


源码github地址:

https://github.com/zhangyu345293721/leetcode

发表评论

表情:
评论列表 (有 0 条评论,236人围观)

还没有评论,来说两句吧...

相关阅读