参考文献 代码随想录
一、最大二叉树
给定一个不重复的整数数组 nums
。 最大二叉树 可以用下面的算法从 nums
递归地构建:
- 创建一个根节点,其值为
nums
中的最大值。 - 递归地在最大值 左边 的 子数组前缀上 构建左子树。
- 递归地在最大值 右边 的 子数组后缀上 构建右子树。
返回 nums
构建的 最大二叉树 。
示例 1:
输入:nums = [3,2,1,6,0,5] 输出:[6,3,5,null,2,0,null,null,1] 解释:递归调用如下所示: - [3,2,1,6,0,5] 中的最大值是 6 ,左边部分是 [3,2,1] ,右边部分是 [0,5] 。 - [3,2,1] 中的最大值是 3 ,左边部分是 [] ,右边部分是 [2,1] 。 - 空数组,无子节点。 - [2,1] 中的最大值是 2 ,左边部分是 [] ,右边部分是 [1] 。 - 空数组,无子节点。 - 只有一个元素,所以子节点是一个值为 1 的节点。 - [0,5] 中的最大值是 5 ,左边部分是 [0] ,右边部分是 [] 。 - 只有一个元素,所以子节点是一个值为 0 的节点。 - 空数组,无子节点。
示例 2:
输入:nums = [3,2,1] 输出:[3,null,2,null,1]
提示:
1 <= nums.length <= 1000
0 <= nums[i] <= 1000
nums
中的所有整数 互不相同
问题分析
构建树要使用前序遍历(必须),然后就使用递归,每次寻找左右 最大的数 ,然后根据这个最大的数寻找到对应的下标分割,以此类推
递归
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution(object):
def constructMaximumBinaryTree(self, nums):
"""
:type nums: List[int]
:rtype: TreeNode
"""
if len(nums) == 1: # 至少要存在一个元素 ,那么就可以了
return TreeNode(nums[0])
maxValue = max(nums)
maxIndex = nums.index(maxValue)
# 中
# 构造树
root = TreeNode(maxValue)
# 左
if maxIndex > 0: # 至少存在 一个 元素
tmp = nums[:maxIndex]
root.left = self.constructMaximumBinaryTree(tmp)
# 右
if maxIndex < len(nums) - 1: # 至少存在 一个 元素
tmp = nums[maxIndex + 1:]
root.right = self.constructMaximumBinaryTree(tmp)
return root
二、合并二叉树
给你两棵二叉树: root1
和 root2
。
想象一下,当你将其中一棵覆盖到另一棵之上时,两棵树上的一些节点将会重叠(而另一些不会)。你需要将这两棵树合并成一棵新二叉树。合并的规则是:如果两个节点重叠,那么将这两个节点的值相加作为合并后节点的新值;否则,不为 null 的节点将直接作为新二叉树的节点。
返回合并后的二叉树。
注意: 合并过程必须从两个树的根节点开始。
示例 1:
输入:root1 = [1,3,2,5], root2 = [2,1,3,null,4,null,7] 输出:[3,4,5,5,4,null,7]
示例 2:
输入:root1 = [1], root2 = [1,2] 输出:[2,2]
提示:
- 两棵树中的节点数目在范围
[0, 2000]
内 -104 <= Node.val <= 104
问题分析
使用哪个遍历顺序都可以,只是前序比较好理解
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution(object):
def mergeTrees(self, root1, root2):
"""
:type root1: TreeNode
:type root2: TreeNode
:rtype: TreeNode
"""
if not root1:
return root2
if not root2:
return root1
root1.val += root2.val # 中
root1.left = self.mergeTrees(root1.left, root2.left) # 左
root1.right = self.mergeTrees(root1.right, root2.right) # 右
return root1
定义新的节点
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution(object):
def mergeTrees(self, root1, root2):
"""
:type root1: TreeNode
:type root2: TreeNode
:rtype: TreeNode
"""
if not root1:
return root2
if not root2:
return root1
root = TreeNode(root1.val + root2.val)
root.left = self.mergeTrees(root1.left, root2.left) # 左
root.right = self.mergeTrees(root1.right, root2.right) # 右
return root
迭代
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution(object):
def mergeTrees(self, root1, root2):
"""
:type root1: TreeNode
:type root2: TreeNode
:rtype: TreeNode
"""
from collections import deque
queen = deque()
if not root1:
return root2
if not root2:
return root1
queen.append((root1, root2))
while queen:
cur1, cur2 = queen.popleft()
if cur1.left and cur2.left: # 只有2个节点都有左节点的时候,在添加
queen.append((cur1.left, cur2.left))
if cur1.right and cur2.right:
queen.append((cur1.right, cur2.right))
# 更新当前节点. 同时改变当前节点的左右孩子.
cur1.val += cur2.val
if not cur1.left and cur2.left:
cur1.left = cur2.left
if not cur1.right and cur2.right:
cur1.right = cur2.right
return root1
给定二叉搜索树(BST)的根节点 root
和一个整数值 val
。
你需要在 BST 中找到节点值等于 val
的节点。 返回以该节点为根的子树。 如果节点不存在,则返回 null
。
示例 1:
输入:root = [4,2,7,1,3], val = 2 输出:[2,1,3]
示例 2:
输入:root = [4,2,7,1,3], val = 5 输出:[]
提示:
- 树中节点数在
[1, 5000]
范围内 1 <= Node.val <= 107
root
是二叉搜索树1 <= val <= 107
问题分析
如果遇到空的或者是等于val的那么就返回root节点如果root.val > val,搜索左子树,如果root.val < val,就搜索右子树,最后如果都没有搜索到,就返回NULL。写递归函数的时候 习惯直接写 searchBST(root.left, val)
,却忘了 递归函数还有返回值。递归函数的返回值是什么? 是 左子树如果搜索到了val,要将该节点返回。 如果不用一个变量将其接住,那么返回值不就没了。所以要 return searchBST(root.left, val)
递归
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution(object):
def searchBST(self, root, val):
"""
:type root: TreeNode
:type val: int
:rtype: TreeNode
"""
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)
迭代
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution(object):
def searchBST(self, root, val):
"""
:type root: TreeNode
:type val: int
:rtype: TreeNode
"""
while root:
if root.val > val:
root = root.left
elif root.val < val:
root = root.right
else:
return root
四、验证二叉搜索树
给你一个二叉树的根节点 root
,判断其是否是一个有效的二叉搜索树。
有效 二叉搜索树定义如下:
- 节点的左
子树
只包含 小于 当前节点的数。 - 节点的右子树只包含 大于 当前节点的数。
- 所有左子树和右子树自身必须也是二叉搜索树。
示例 1:
输入:root = [2,1,3] 输出:true
示例 2:
输入:root = [5,1,4,null,null,3,6] 输出:false 解释:根节点的值是 5 ,但是右子节点的值是 4 。
提示:
- 树中节点数目范围在
[1, 104]
内 -231 <= Node.val <= 231 - 1
问题分析
利用搜索二叉树的特性,判断它是否是,中序遍历一定是升序的
递归
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution(object):
def __init__(self):
self.result = []
def isValidBST(self, root):
"""
:type root: TreeNode
:rtype: bool
"""
# 搜索二叉树的中序遍历一定是有有序的
if not root:
return
self.isValidBST(root.left)
self.result.append(root.val)
self.isValidBST(root.right)
for i in range(1, len(self.result)):
if self.result[i] <= self.result[i - 1]:
return False
return True
递归优化(就是当前节点与之前的左对比)
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution(object):
def __init__(self):
self.result = []
self.pre = None
def isValidBST(self, root):
"""
:type root: TreeNode
:rtype: bool
"""
# 搜索二叉树的中序遍历一定是有有序的
return self.dfs(root)
def dfs(self, root):
if not root:
return True
left = self.isValidBST(root.left)
if self.pre and root.val <= self.pre.val:
return False
self.pre = root
right = self.isValidBST(root.right)
return right and left
迭代
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution(object):
def __init__(self):
self.result = []
def isValidBST(self, root):
"""
:type root: TreeNode
:rtype: bool
"""
# 搜索二叉树的中序遍历一定是有有序的
from collections import deque
stack = deque()
stack.append(root)
while stack:
cur = stack.pop()
if cur:
if cur.right:
stack.append(cur.right)
stack.append(cur)
stack.append(None)
if cur.left:
stack.append(cur.left)
else:
cur = stack.pop()
self.result.append(cur.val)
for i in range(1, len(self.result)):
if self.result[i] <= self.result[i - 1]:
return False
return True