leetcode树之二叉树剪枝

669、修剪二叉搜索树

给你二叉搜索树的根节点 root ，同时给定最小边界low 和最大边界 high。通过修剪二叉搜索树，使得所有节点的值在[low, high]中。修剪树 不应该 改变保留在树中的元素的相对结构 (即，如果没有被移除，原有的父代子代关系都应当保留)。 可以证明，存在 唯一的答案 。

所以结果应当返回修剪好的二叉搜索树的新的根节点。注意，根节点可能会根据给定的边界发生改变。

示例1：

输入：root = [1,0,2], low = 1, high = 2
输出：[1,null,2]

示例2：

输入：root = [3,0,4,null,2,null,null,1], low = 1, high = 3
输出：[3,2,null,1]

---

思路：

二叉搜索树左子树的值小于根节点值，右子树的值大于根节点的值，然后根据这个规则，如果根节点不满足最小边界，那么整个左子树其实也不满足，那么就只需要去判断根节点的右子树，同理右子树也是一样。

class Solution:
    def trimBST(self, root: Optional[TreeNode], low: int, high: int) -> Optional[TreeNode]:
        if not root: return root
        # 如果根节点的值不满足最小边界，那么返回剪枝后的右子树，同理不满足最大边界，返回剪枝后的左子树
        if root.val < low: return self.trimBST(root.right, low, high)
        if root.val > high: return self.trimBST(root.left, low, high)
        # 如果根节点满足边界值，那么只需要修剪左右子树即可
        root.left = self.trimBST(root.left, low, high)
        root.right = self.trimBST(root.right, low, high)
        return root

814、二叉树剪枝

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

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

节点 node 的子树为 node 本身加上所有 node 的后代。

示例1：

输入：root = [1,null,0,0,1]
输出：[1,null,0,null,1]

示例2：

输入：root = [1,0,1,0,0,0,1]
输出：[1,null,1,null,1]

示例3：

输入：root = [1,1,0,1,1,0,1,0]
输出：[1,1,0,1,1,null,1]

---

思路：

使用递归迭代删除节点为0的树节点

class Solution:
    def pruneTree(self, root: Optional[TreeNode]) -> Optional[TreeNode]:
        def dfs(root):
            if not root: return None
            # 如果没有左右子节点，说明该结点是叶子结点，并且值为0，那么删除即可
            if not root.left and not root.right and root.val == 0:
                return None
            left = dfs(root.left)
            right = dfs(root.right)
            # 如果没有左右子树，说明可能左右子树都被剪枝了，这个时候如果root值为0，那么删除即可
            if not left and not right and root.val == 0:
                return None
            # 更新左右子树的值
            root.left = left
            root.right = right
            return root
        # 返回新的剪枝结果
        return dfs(root)