剑指 Offer II 047. 二叉树剪枝

后序遍历 递归就很绕。。

 func pruneTree(_ root: TreeNode?) -> TreeNode? {
      
        if root == nil {
            return root
        }
        let tempNode = root!

        tempNode.left = pruneTree(tempNode.left)
        tempNode.right = pruneTree(tempNode.right)
        
        if tempNode.left == nil && tempNode.right == nil && tempNode.val == 0 {
            return nil
        }
        return tempNode
        
    }