本文主要是介绍【leetcode刷刷】669. 修剪二叉搜索树 、108.将有序数组转换为二叉搜索树 、538.把二叉搜索树转换为累加树,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
669. 修剪二叉搜索树
- 一些递归,有点点绕,但画了一下示意图就差不多能理清
class Solution:def trimBST(self, root: Optional[TreeNode], low: int, high: int) -> Optional[TreeNode]:# 修剪树——保留原来的父代子代关系# 遇到的第一个小于low的节点,这个节点的左子节点全部删除,右子节点部分保留# 遇到的第一个大于high的节点,这个节点的左子节点部分保留,右子节点全部删除dummy = TreeNode(high, left=root)self.traversal(dummy, low, high)return dummy.leftdef traversal(self, root, low, high):if not root: return if root.val < low:return self.traversal(root.right, low, high)elif root.val > high:return self.traversal(root.left, low, high)else:root.left = self.traversal(root.left, low, high)root.right = self.traversal(root.right, low, high)return root
- 不知道为什么最近陷入这个dummy的怪圈了。其实由于是递归,因此return递归返回的值就行了,返回的就是root。
class Solution:def trimBST(self, root: Optional[TreeNode], low: int, high: int) -> Optional[TreeNode]:return self.traversal(root, low, high)def traversal(self, root, low, high):if not root: return if root.val < low:return self.traversal(root.right, low, high)elif root.val > high:return self.traversal(root.left, low, high)else:root.left = self.traversal(root.left, low, high)root.right = self.traversal(root.right, low, high)return root
108.将有序数组转换为二叉搜索树
class Solution:def sortedArrayToBST(self, nums: List[int]) -> Optional[TreeNode]:# 选取中间值?n = len(nums)if n == 0: return Nonemid_node = nums[n//2]root = TreeNode(mid_node)root.left = self.sortedArrayToBST(nums[:n//2])root.right = self.sortedArrayToBST(nums[n//2+1:])return root
538.把二叉搜索树转换为累加树
- 反中序的递归,累加。其实很简单,奈何自己递归还是有点想不明白
class Solution:def convertBST(self, root: Optional[TreeNode]) -> Optional[TreeNode]:# 中序,最右第一个处理,然后中,然后左self.pre = 0self.traversal(root)return rootdef traversal(self, cur):if not cur: return 0self.traversal(cur.right)cur.val += self.preself.pre = cur.valself.traversal(cur.left)
这篇关于【leetcode刷刷】669. 修剪二叉搜索树 、108.将有序数组转换为二叉搜索树 、538.把二叉搜索树转换为累加树的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!