[LC814]二叉树剪枝

题目

题目地址

分析

这道题符合递归的性质，对于当前的节点node，当且仅当其左右孩子都为不包含1的子树，且node.val=1时，node所在的子树才符合“不包含1的子树”这一定义。那么很自然的，我们可以采取树的后序处理，递归的处理上述条件，具体代码如下。

代码

Golang代码

/*
 * @lc app=leetcode.cn id=814 lang=golang
 *
 * [814] 二叉树剪枝
 */

// @lc code=start
/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func pruneTree(root *TreeNode) *TreeNode {
	// 若为空节点，直接返回nil
	if root == nil {
		return nil
	}
	// 后序操作
	root.Left = pruneTree(root.Left)
	root.Right = pruneTree(root.Right)
	// 这里的nil表示子树全为0
	if root.Left != nil || root.Right != nil {
		return root
	}
	// 若左右子树都全为0，则还需判断当前节点的值
	if root.Val == 0 {
		return nil
	}
	return root
}

// @lc code=end