给定二叉树根结点 root ，此外树的每个结点的值要么是 0，要么是 1。

返回移除了所有不包含 1 的子树的原二叉树。

( 节点 X 的子树为 X 本身，以及所有 X 的后代。)

示例1:
输入: [1,null,0,0,1]
输出: [1,null,0,null,1]
 
解释: 
只有红色节点满足条件“所有不包含 1 的子树”。
右图为返回的答案。


示例2:
输入: [1,0,1,0,0,0,1]
输出: [1,null,1,null,1]



示例3:
输入: [1,1,0,1,1,0,1,0]
输出: [1,1,0,1,1,null,1]

说明:

  • 给定的二叉树最多有 100 个节点。
  • 每个节点的值只会为 0 或 1 。

 

思路：

想了半天，后来看网上的思路，原来只是递归，但是退出条件有点难确定。直接看代码注释。

 public TreeNode pruneTree(TreeNode root) {
		if (root == null) {      //递归退出条件
			return null;
		}
		if (isLeafRoot(root)) { // 如果为叶子 递归退出条件
			return root.val == 0 ? null : root;
		}
		// 递归  每一个节点的值由dfs之后的结果决定
		root.left = pruneTree(root.left);
		root.right = pruneTree(root.right);
		return isLeafRoot(root) && root.val == 0 ? null : root;
	}

	// 判断是否为叶子
	private static boolean isLeafRoot(TreeNode root) {
		if( root.left == null && root.right == null)
             return true;
        return false;
	}