给定二叉搜索树(BST)的根节点 root 和要插入树中的值 value ,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据 保证 ,新值和原始二叉搜索树中的任意节点值都不同。
注意,可能存在多种有效的插入方式,只要树在插入后仍保持为二叉搜索树即可。 你可以返回 任意有效的结果 。
示例 1:
输入:root = [4,2,7,1,3], val = 5
输出:[4,2,7,1,3,5]
解释:另一个满足题目要求可以通过的树是:
示例 2:
输入:root = [40,20,60,10,30,50,70], val = 25
输出:[40,20,60,10,30,50,70,null,null,25]
示例 3:
输入:root = [4,2,7,1,3,null,null,null,null,null,null], val = 5
输出:[4,2,7,1,3,5]
思路:
采用经典的插入方法,使整体操作变化最小。
直接深拷贝构造新树,然后修改 寻找到合适的叶位置后插入新节点,
这样只需要在原树的某个叶节点处延伸一个节点
# 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 insertIntoBST(self, root: TreeNode, val: int) -> TreeNode:
res = copy.deepcopy(root)
cur_node = res
# 如果二叉搜索树为空树,用val构造二叉树节点作为根节点并返回
if cur_node == None:
return TreeNode(val)
while cur_node:
if val > cur_node.val:
# 走向右子树
if cur_node.right != None:
cur_node = cur_node.right
else:
# 右子树为空,则找到右子树插入位置
cur_node.right = TreeNode(val)
return res
elif val < cur_node.val:
if cur_node.left != None:
cur_node = cur_node.left
else:
cur_node.left = TreeNode(val)
return res
70. Leetcode 701. 二叉搜索树中的插入操作 (二叉搜索树-基本操作类)
最新推荐文章于 2024-08-25 17:36:33 发布