669. 修剪二叉搜索树
1.题目
给你二叉搜索树的根节点 root ,同时给定最小边界low 和最大边界 high。通过修剪二叉搜索树,使得所有节点的值在[low, high]中。修剪树 不应该 改变保留在树中的元素的相对结构 (即,如果没有被移除,原有的父代子代关系都应当保留)。 可以证明,存在 唯一的答案 。
2.复习前
难点:对于找到不在区间内的点如何处理,不能直接返回其左节点或右节点!
3.文章讲解
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
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) #既然要继续分析root.right,则应继续加入递归中
if root.val > high:
return self.trimBST(root.left, low, high)
if root.val >= low and root.val <= high:
root.left = self.trimBST(root.left, low, high)
root.right = self.trimBST(root.right, low, high)
return root
总结:对于有返回值的点,如需进一步处理返回值,应该让其继续递归(在相同条件下进行递归判断)
108.将有序数组转换为二叉搜索树
1.题目
给你一个整数数组 nums ,其中元素已经按 升序 排列,请你将其转换为一棵 高度平衡 二叉搜索树。高度平衡 二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过 1 」的二叉树。
2.实现
开始时被高度平衡吓到了,其实先想想如何构造BST,即可顺利避免判断高度平衡
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
def sortedArrayToBST(self, nums: List[int]) -> Optional[TreeNode]:
# 满足左根右的节点安排,于是取最中间为根节点,左边为左子树,右边为右子树
n = len(nums)
if n == 0:
return None
root = TreeNode(nums[n // 2])
root.left = self.sortedArrayToBST(nums[: n//2])
root.right = self.sortedArrayToBST(nums[n//2 + 1:])
return root
3.文章讲解
538.把二叉搜索树转换为累加树
1.题目
给出二叉 搜索 树的根节点,该树的节点值各不相同,请你将其转换为累加树(Greater Sum Tree),使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。
看文字有点头疼,建议看实例图
2.实现
用到了双指针(这种前后相关的一定要把握好双指针的方法!)
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
def __init__(self):
self.pre = None
def convertBST(self, root: Optional[TreeNode]) -> Optional[TreeNode]:
if not root:
return root
root.right = self.convertBST(root.right)
if self.pre:
root.val += self.pre.val
self.pre = root
root.left = self.convertBST(root.left)
return root
3.文章讲解