代码随想录算法训练营第二十三日｜LC669. 修剪二叉搜索树 LC108.将有序数组转换为二叉搜索树

最新推荐文章于 2024-05-20 22:00:16 发布

EdisonW8

最新推荐文章于 2024-05-20 22:00:16 发布

阅读量121

点赞数

文章标签：算法 leetcode 职场和发展

本文链接：https://blog.csdn.net/EdisonW8/article/details/127932116

版权

LC669修剪二叉搜索树：

class Solution:
    def trimBST(self, root: Optional[TreeNode], low: int, high: int) -> Optional[TreeNode]:
        if not root:
            return None
        if root.val < low:
            # 若当前root节点小于左界：只考虑其右子树，用于替代更新后的其本身，抛弃其左子树整体
            return self.trimBST(root.right, low, high)
        if root.val > high:
            # 若当前root节点大于右界：只考虑其左子树，用于替代更新后的其本身，抛弃其右子树整体
            return self.trimBST(root.left, low, high)
        if low <= root.val <= high:
            root.left = self.trimBST(root.left, low, high)
            root.right = self.trimBST(root.right, low, high)
            # 返回更新后的剪枝过的当前节点root
            return root

LC108将有序数组转换为二叉搜索树

class Solution:
    def sortedArrayToBST(self, nums: List[int]) -> Optional[TreeNode]:
        root = self.traversal(nums, 0, len(nums)-1)
        return root

    def traversal(self, nums: List[int], left:int, right:int) -> TreeNode:
        if left > right:
            return None
        # 确定左右界的中心，防越界
        mid = left + (right - left) // 2
        mid_root = TreeNode(nums[mid])
        # 构建以左右界的中心为分割点的左右子树
        mid_root.left = self.traversal(nums, left, mid-1)
        mid_root.right = self.traversal(nums, mid + 1, right)

        # 返回由被传入的左右界定义的某子树的根节点
        return mid_root