leetcode第337题打家劫舍III
**思路:**动态规划,分别记录偷和没偷的情况,比较。
- **后序遍历的由来:**需要知道子节点偷没偷,子节点偷了,当前节点就不能偷,当前节点要偷,左右子节点都不能偷。和监控二叉树挺像的,感觉。
- 每个节点只有偷和没偷两种状态,所有dp的初始化为第三,在递归的过程中,栈会保存每一层递归的参数。
- 再递归的过程中,无论偷或者不偷,最大的利润为0
- 举例推导dp数组,这一环是很重要的,可以正确验证状态转移方程的正确性。
//这种题目,我好想明白了,但是又好像没有明白,我应该怎么办呢?
class Solution {
public int rob(TreeNode root) {
int[] res = DFS(root);
return Math.max(res[0],res[1]);
}
int[] DFS(TreeNode root){
if(root == null){
return new int[2];
}
int[] left = DFS(root.left);
int[] right = DFS(root.right);
int[] dp = new int[2];
//我没偷
dp[0] =Math.max(left[0],left[1]) + Math.max(right[0],right[1]);
//偷了
dp[1] = root.val + left[0] + right[0];
return dp;
}
}