337. 打家劫舍 III

港控/mmm° 2022-11-26 09:57 216阅读 0赞

题目:

337. 打家劫舍 III
在这里插入图片描述

题解:动态规划

在这里插入图片描述

1. 解释一:

在这里插入图片描述

  1. root[0] = Math.max(rob(root.left)[0], rob(root.left)[1]) + Math.max(rob(root.right)[0], rob(root.right)[1])
  2. root[1] = rob(root.left)[0] + rob(root.right)[0] + root.val;

2. 解释二:

在这里插入图片描述

3. 解释三:

在这里插入图片描述

代码:动态规划

  1. public class code337 {
  2. // 树的后序遍历
  3. public static int rob(TreeNode root) {
  4. // 计算当前节点偷与不偷所能获得收益,结果存在数组res中
  5. int res[] = dfs(root);
  6. // 根据题意可知需取其中最大的
  7. return Math.max(res[0], res[1]);
  8. }
  9. public static int[] dfs(TreeNode root)
  10. {
  11. // 如果当前节点为空节点,则其结果为0
  12. if(root == null)
  13. {
  14. int temp[] = { 0, 0 };
  15. return temp;
  16. }
  17. // 分类讨论的标准是:当前结点偷或者不偷
  18. // 由于需要后序遍历,所以先计算左右子结点,然后计算当前结点的状态值
  19. // 计算当前节点左儿子偷与不偷所能获得的收益
  20. int left[] = dfs(root.left);
  21. // 计算当前节点右儿子偷与不偷所能获得的收益
  22. int right[] = dfs(root.right);
  23. // dp[0]:以当前 node 为根结点的子树能够偷取的最大价值,规定 node 结点不偷
  24. // dp[1]:以当前 node 为根结点的子树能够偷取的最大价值,规定 node 结点偷
  25. int dp[] = new int[2];
  26. // 不偷,下家可偷可不偷,取决于收益大小
  27. // 不偷当前节点所能获得的最大收益 = 左儿子所能获得的最大收益 + 右儿子所能获得的最大收益
  28. // 儿子所能获得最大收益为:max(不偷当前儿子所能获得最大收益,偷当前儿子所能获得最大收益)
  29. dp[0] = Math.max(left[0], left[1]) + Math.max(right[0], right[1]);
  30. // 偷,下家就不能偷了
  31. // 偷当前节点所能获得的最大收益= 偷当前节点的钱 + 不偷左儿子所获得的钱 +不偷右儿子所获得的钱
  32. dp[1] = root.val + left[0] + right[0];
  33. return dp;
  34. }
  35. public static void main(String[] args) {
  36. Integer nums1[] = { 3, 2, 3, null, 3, null, 1 };
  37. TreeNode root1 = ConstructTree.constructTree(nums1);
  38. TreeOperation.show(root1);
  39. System.out.println("***************************************");
  40. int res1 = rob(root1);
  41. System.out.println(res1);
  42. System.out.println("***************************************");
  43. Integer nums2[] = { 3, 4, 5, 1, 3, null, 1 };
  44. TreeNode root2 = ConstructTree.constructTree(nums2);
  45. TreeOperation.show(root2);
  46. System.out.println("***************************************");
  47. int res2 = rob(root2);
  48. System.out.println(res2);
  49. System.out.println("***************************************");
  50. Integer nums3[] = { 2, 1, 3, null, 4 };
  51. TreeNode root3 = ConstructTree.constructTree(nums3);
  52. TreeOperation.show(root3);
  53. System.out.println("***************************************");
  54. int res3 = rob(root3);
  55. System.out.println(res3);
  56. System.out.println("***************************************");
  57. }
  58. }

参考:

  1. 打家劫舍 III
  2. 三种方法解决树形动态规划问题-从入门级代码到高效树形动态规划代码实现
  3. 树形 dp 入门问题(理解「无后效性」和「后序遍历」)
  4. 通用思路团灭打家劫舍问题
  5. 三种解法+多图演示 337. 打家劫舍 III
  6. 337. 打家劫舍 III

发表评论

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

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

相关阅读