337. 打家劫舍 III（中等）

思路：

动态规划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