题目链接:654. 最大二叉树 - 力扣(Leetcode)617. 合并二叉树 - 力扣(Leetcode)700. 二叉搜索树中的搜索 - 力扣(Leetcode)98. 验证二叉搜索树 - 力扣(Leetcode)
654 最大二叉树
给定一个不重复的整数数组 nums
。 最大二叉树 可以用下面的算法从 nums
递归地构建:
- 创建一个根节点,其值为
nums
中的最大值。 - 递归地在最大值 左边 的 子数组前缀上 构建左子树。
- 递归地在最大值 右边 的 子数组后缀上 构建右子树。
返回 nums
构建的 最大二叉树 。
输入:nums = [3,2,1,6,0,5] 输出:[6,3,5,null,2,0,null,null,1]
思路:找到最大值分割左右节点区间后,再各自找最大值分割。
自写ac——递归
class Solution:
def constructMaximumBinaryTree(self, nums: List[int]) -> Optional[TreeNode]:
if not nums:
return None
max_num = max(nums)
location = nums.index(max_num)
root = TreeNode(max_num)//构建二叉树
leftPart = nums[:location]
rightPart = nums[location + 1:]
root.left = self.constructMaximumBinaryTree(leftPart)
root.right = self.constructMaximumBinaryTree(rightPart)
return root
617 合并二叉树
给你两棵二叉树: root1
和 root2
。
想象一下,当你将其中一棵覆盖到另一棵之上时,两棵树上的一些节点将会重叠(而另一些不会)。你需要将这两棵树合并成一棵新二叉树。合并的规则是:如果两个节点重叠,那么将这两个节点的值相加作为合并后节点的新值;否则,不为 null 的节点将直接作为新二叉树的节点。
返回合并后的二叉树。
注意: 合并过程必须从两个树的根节点开始。
输入:root1 = [1,3,2,5], root2 = [2,1,3,null,4,null,7] 输出:[3,4,5,5,4,null,7]
本题需要同时操作两棵二叉树,但要注意的是递归函数也是需要同时传两个参数的。
自写——前序
class Solution:
def mergeTrees(self, root1: Optional[TreeNode], root2: Optional[TreeNode]) -> Optional[TreeNode]:
if not root1 and not root2:
return None
if not root1 and root2:
return root2
if root1 and not root2:
return root1
root = TreeNode(root1.val + root2.val)
if root1.left and root2.left:
root.left = root1.left.val + root2.left.val
if not root1.left and root2.left:
root.left = root2.left.val
if root1.left and not root2.left:
root.left = root1.left.val
left1, left2 = self.mergeTrees(root1.left, root2.left)
if root1.right and root2.right:
root.right = root1.right.val + root2.right.val
if not root1.right and root2.right:
root.right = root2.right.val
if root1.right and not root2.right:
root.right = root1.right.val
right1, right2 = self.mergeTrees(root1.right, root2.right)
return root
问题:太过复杂,且费时费空间
参考代码:
class Solution:
def mergeTrees(self, root1: TreeNode, root2: TreeNode) -> TreeNode:
# 递归终止条件:
# 但凡有一个节点为空, 就立刻返回另外一个. 如果另外一个也为None就直接返回None.
if not root1:
return root2
if not root2:
return root1
# 上面的递归终止条件保证了代码执行到这里root1, root2都非空.
root1.val += root2.val # 中
root1.left = self.mergeTrees(root1.left, root2.left) #左
root1.right = self.mergeTrees(root1.right, root2.right) # 右
return root1 #节省时间
700 二叉搜索树中的搜索
给定二叉搜索树(BST)的根节点 root
和一个整数值 val
。
你需要在 BST 中找到节点值等于 val
的节点。 返回以该节点为根的子树。 如果节点不存在,则返回 null
。
输入:root = [4,2,7,1,3], val = 2 输出:[2,1,3]
自写
class Solution:
def searchBST(self, root: Optional[TreeNode], val: int) -> Optional[TreeNode]:
if not root:
return None
if root.val == val:
return root
elif root.val > val:
self.searchBST(root.left, val)
elif root.val < val:
self.searchBST(root.right, val)
递归没有返回值;要好好利用搜索树左大右小的特性。
参考代码
class Solution:
def searchBST(self, root: TreeNode, val: int) -> TreeNode:
# 为什么要有返回值:
# 因为搜索到目标节点就要立即return,
# 这样才是找到节点就返回(搜索某一条边),如果不加return,就是遍历整棵树了。
if not root or root.val == val:
return root
if root.val > val:
return self.searchBST(root.left, val)
if root.val < val:
return self.searchBST(root.right, val)
98 验证二叉搜索树
给你一个二叉树的根节点 root
,判断其是否是一个有效的二叉搜索树。
有效 二叉搜索树定义如下:
- 节点的左子树只包含 小于 当前节点的数。
- 节点的右子树只包含 大于 当前节点的数。
- 所有左子树和右子树自身必须也是二叉搜索树。
输入:root = [5,1,4,null,null,3,6] 输出:false 解释:根节点的值是 5 ,但是右子节点的值是4.
二叉搜索树特性:中序遍历时,元素有序
注意,以示例为例,即使右子节点为7,6改为8,那也是false,因为3小于根节点5,而3属于右子树,二叉搜索树需要右子树全都大于根节点,左子树全都小于根节点。
参考代码
class Solution:
def isValidBST(self, root: TreeNode) -> bool:
# 规律: BST的中序遍历节点数值是从小到大.
cur_max = -float("INF")
def __isValidBST(root: TreeNode) -> bool:
nonlocal cur_max
if not root:
return True
is_left_valid = __isValidBST(root.left)
if cur_max < root.val:
cur_max = root.val
else:
return False
is_right_valid = __isValidBST(root.right)
return is_left_valid and is_right_valid
return __isValidBST(root)