337. 打家劫舍 III
题目:
337. 打家劫舍 III
题解:动态规划
1. 解释一:
root[0] = Math.max(rob(root.left)[0], rob(root.left)[1]) + Math.max(rob(root.right)[0], rob(root.right)[1])
root[1] = rob(root.left)[0] + rob(root.right)[0] + root.val;
2. 解释二:
3. 解释三:
代码:动态规划
public class code337 {
// 树的后序遍历
public static int rob(TreeNode root) {
// 计算当前节点偷与不偷所能获得收益,结果存在数组res中
int res[] = dfs(root);
// 根据题意可知需取其中最大的
return Math.max(res[0], res[1]);
}
public static int[] dfs(TreeNode root)
{
// 如果当前节点为空节点,则其结果为0
if(root == null)
{
int temp[] = { 0, 0 };
return temp;
}
// 分类讨论的标准是:当前结点偷或者不偷
// 由于需要后序遍历,所以先计算左右子结点,然后计算当前结点的状态值
// 计算当前节点左儿子偷与不偷所能获得的收益
int left[] = dfs(root.left);
// 计算当前节点右儿子偷与不偷所能获得的收益
int right[] = dfs(root.right);
// dp[0]:以当前 node 为根结点的子树能够偷取的最大价值,规定 node 结点不偷
// dp[1]:以当前 node 为根结点的子树能够偷取的最大价值,规定 node 结点偷
int dp[] = new int[2];
// 不偷,下家可偷可不偷,取决于收益大小
// 不偷当前节点所能获得的最大收益 = 左儿子所能获得的最大收益 + 右儿子所能获得的最大收益
// 儿子所能获得最大收益为:max(不偷当前儿子所能获得最大收益,偷当前儿子所能获得最大收益)
dp[0] = Math.max(left[0], left[1]) + Math.max(right[0], right[1]);
// 偷,下家就不能偷了
// 偷当前节点所能获得的最大收益= 偷当前节点的钱 + 不偷左儿子所获得的钱 +不偷右儿子所获得的钱
dp[1] = root.val + left[0] + right[0];
return dp;
}
public static void main(String[] args) {
Integer nums1[] = { 3, 2, 3, null, 3, null, 1 };
TreeNode root1 = ConstructTree.constructTree(nums1);
TreeOperation.show(root1);
System.out.println("***************************************");
int res1 = rob(root1);
System.out.println(res1);
System.out.println("***************************************");
Integer nums2[] = { 3, 4, 5, 1, 3, null, 1 };
TreeNode root2 = ConstructTree.constructTree(nums2);
TreeOperation.show(root2);
System.out.println("***************************************");
int res2 = rob(root2);
System.out.println(res2);
System.out.println("***************************************");
Integer nums3[] = { 2, 1, 3, null, 4 };
TreeNode root3 = ConstructTree.constructTree(nums3);
TreeOperation.show(root3);
System.out.println("***************************************");
int res3 = rob(root3);
System.out.println(res3);
System.out.println("***************************************");
}
}
参考:
- 打家劫舍 III
- 三种方法解决树形动态规划问题-从入门级代码到高效树形动态规划代码实现
- 树形 dp 入门问题(理解「无后效性」和「后序遍历」)
- 通用思路团灭打家劫舍问题
- 三种解法+多图演示 337. 打家劫舍 III
- 337. 打家劫舍 III
还没有评论,来说两句吧...