701. 二叉搜索树中的插入操作

最新推荐文章于 2024-05-16 21:56:43 发布

小杜在学习

最新推荐文章于 2024-05-16 21:56:43 发布

阅读量154

点赞数

CC 4.0 BY-SA版权

分类专栏：算法刷题文章标签： leetcode 数据结构算法

本文链接：https://blog.csdn.net/weixin_44571635/article/details/124991327

算法刷题专栏收录该内容

52 篇文章

订阅专栏

本文介绍了一种在二叉搜索树中插入新值的方法。通过迭代法遍历树并记录父节点，找到合适的位置进行插入，确保树的二叉搜索特性不变。

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

701. 二叉搜索树中的插入操作

每日几道leetcode刷刷题！
传送门

题目描述

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

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

示例 1:

代码

在迭代法遍历的过程中，需要记录一下当前遍历的节点的父节点，这样才能做插入节点的操作，所以要用pre和cur记录，当cur 遍历到叶子节点时，此时需要终止遍历，cur = None，但pre还在叶子节点那（相当于父节点），这样就可以赋值了。

class Solution:
    def insertIntoBST(self, root: TreeNode, val: int) -> TreeNode:
        if not root:
            node = TreeNode(val)
            return node
        cur = root
        pre = root
        while cur:
            pre = cur
            if cur.val > val:
                cur = cur.left
            else:
                cur = cur.right
        #直到cur遍历到叶子节点，然后赋值
        node = TreeNode(val)
        if pre.val > val:
            pre.left = node
        else:
            pre.right = node
        return root