思路:
动态规划dp,后序遍历(消除后效性)
代码:
class Solution {
public int rob(TreeNode root) {
//注意dfs返回的参数是int[],2个值(根节点偷/不偷),还要比较哪一种比较大
int[] res=dfs(root);
return Math.max(res[0],res[1]);
}
private int[] dfs(TreeNode root){
if(root==null){
return new int[]{0,0};
}
//后序遍历:左右根
//当前节点偷不偷由子节点偷不偷来决定
int[] left=dfs(root.left);
int[] right=dfs(root.right);
//定义2种状态:
//0不偷
//1偷
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;
}
}
分解:
1)本节点偷不偷要看左右子节点偷不偷,从这句话来看,要用二叉树的后序遍历,来消除后效性
2)状态转移方程:
i)根节点不偷:左子节点偷/不偷,右子节点偷/不偷
dp[0]=Math.max(left[0],left[1])+Math.max(right[0],right[1]);
ii)根节点偷:左右子节点都不偷,要记得加上根节点的值
dp[1]=root.val+left[0]+right[0];
复杂度分析:
时间复杂度:O(N)后序遍历把所有节点都遍历一遍
空间复杂度:O(N)栈的深度最差为N