【代码随想录day21】二叉搜索树中的插入操作

最新推荐文章于 2025-08-25 22:11:31 发布

原创最新推荐文章于 2025-08-25 22:11:31 发布 · 418 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #算法 #leetcode

LeetCode 同时被 3 个专栏收录

112 篇文章

订阅专栏

数据结构

97 篇文章

订阅专栏

代码随想录

61 篇文章

订阅专栏

该文章介绍如何在保持二叉搜索树性质不变的情况下，插入一个新值。方法是通过递归找到合适的位置进行插入，如果当前节点为空，则创建新节点；若新值小于当前节点值，向左子树递归，否则向右子树递归。

题目

给定二叉搜索树（BST）的根节点 root 和要插入树中的值 value ，将值插入二叉搜索树。返回插入后二叉搜索树的根节点。输入数据保证，新值和原始二叉搜索树中的任意节点值都不同。

注意，可能存在多种有效的插入方式，只要树在插入后仍保持为二叉搜索树即可。你可以返回任意有效的结果。

思路

核心思路，其实无需调整整颗二叉排序树，只需要找到插入的位置即可。

# 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: Optional[TreeNode], val: int) -> Optional[TreeNode]:
        def solve(root, val):
            if not root:
                return 
            if not root.left and not root.right:
                if root.val>val:
                    root.left = TreeNode(val)
                else:
                    root.right = TreeNode(val)
                return 
            elif not root.left and root.val>val:
                root.left = TreeNode(val)
                return
            elif not root.right and root.val<val:
                root.right = TreeNode(val)
                return

            if val<root.val:
                solve(root.left, val)
            if val>root.val:
                solve(root.right, val)
        if not root:
            return TreeNode(val)
        solve(root, val)
        return root