Day 20
二叉树完结!
669. 修剪二叉搜索树(中等)
给你二叉搜索树的根节点 root ,同时给定最小边界low 和最大边界 high。通过修剪二叉搜索树,使得所有节点的值在[low, high]中。修剪树不应该改变保留在树中的元素的相对结构 (即,如果没有被移除,原有的父代子代关系都应当保留)。可以证明,存在唯一的答案。所以结果应当返回修剪好的二叉搜索树的新的根节点。注意,根节点可能会根据给定的边界发生改变。
解:注意,修剪的时候不是单纯删除就可以的,还要考虑删除节点的右孩子放哪里,如图:
递归时,要考虑三种情况:一,root.val < low,则递归右子树,并返回右子树符合条件的头结点。二,root.val > high,则递归左子树,并返回左子树符合条件的头结点。此时得到的是两个在范围内的左右节点。那么当root.val在范围中时,他的左右节点就将改编成是在递归中找到的符合条件的子节点,因而不符合的部分就在递归过程中被跳过了。
class Solution:
def trimBST(self, root: Optional[TreeNode], low: int, high: int) -> Optional[TreeNode]:
if not root:
return None
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
108. 将有序数组转换为二叉搜索树
给你一个整数数组 nums ,其中元素已经按升序排列,请你将其转换为一棵高度平衡二叉搜索树。高度平衡二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过1」的二叉树。
解:高度平衡,就取中点为根节点然后分割左右子树
class Solution:
def sortedArrayToBST(self, nums: List[int]) -> Optional[TreeNode]:
def traversal(nums, left, right):
if left>right:
return None
mid = left + (right - left) // 2 # 防溢出
root = TreeNode(nums[mid])
root.left = traversal(nums, left, mid-1)
root.right = traversal(nums, mid+1, right)
return root
root = traversal(nums, 0, len(nums)-1)
return root
538. 把二叉搜索树转换为累加树(中等)
给出二叉搜索树的根节点,该树的节点值各不相同,请你将其转换为累加树(Greater Sum Tree),使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。
解:感觉还蛮简单的,就是从右开始,从下往上加,相当于右中左的遍历。注意设定一个pre来保存root的前一个结点的值
class Solution:
def convertBST(self, root: Optional[TreeNode]) -> Optional[TreeNode]:
pre = 0
def traversal(root):
nonlocal pre
if not root:
return
traversal(root.right)
root.val += pre
pre = root.val
traversal(root.left)
traversal(root)
return root