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

小小的香辛料

于 2023-07-27 17:46:09 发布

阅读量371

点赞数

分类专栏：代码随想录 LeetCode 数据结构文章标签： python 算法 leetcode

本文链接：https://blog.csdn.net/doubleguy/article/details/131965472

版权

LeetCode 同时被 3 个专栏收录

112 篇文章

订阅专栏

数据结构

97 篇文章

订阅专栏

代码随想录

61 篇文章

订阅专栏

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

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目

给定二叉搜索树（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