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

最新推荐文章于 2024-10-01 05:04:32 发布

996冲冲冲

最新推荐文章于 2024-10-01 05:04:32 发布

阅读量94

点赞数

分类专栏：二叉搜索树文章标签： leetcode 数据结构

本文链接：https://blog.csdn.net/weixin_44392348/article/details/124992309

版权

二叉搜索树专栏收录该内容

17 篇文章 0 订阅

订阅专栏

先找到插入位置，之后需要连接到树上

def insertIntoBST(self, root, val):
        """
        :type root: TreeNode
        :type val: int
        :rtype: TreeNode
        """
        if not root:
            return TreeNode(val)
        if root.val > val:
            root.left = self.insertIntoBST(root.left,val)
        if root.val < val:
            root.right = self.insertIntoBST(root.right,val)
        return root

不直接返回值

def insertIntoBST(self, root, val):
        parent = None
        if not root: 
            return TreeNode(val)
        def traverse(cur, val): 
            global parent # 在函数运行的同时把新节点插入到该被插入的地方. 
            if not cur:   #这里说明找到了插入的位置，然后判断一下是放在左边还是右边
                new_node = TreeNode(val)
                if parent.val < val: 
                    parent.right = new_node
                else: 
                    parent.left = new_node
                return 
            parent = cur    # 这里的parent记录要插入的位置，是cur的父节点.
            if cur.val < val: 
                traverse(cur.right, val)
            else: 
                traverse(cur.left, val)
        traverse(root, val)
        return root

迭代

def insertIntoBST(self, root, val):
        """
        :type root: TreeNode
        :type val: int
        :rtype: TreeNode
        """
        if not root:
            return TreeNode(val)
        parent = None
        roott = root
        while root:
            if root.val > val:
                parent = root
                root = root.left
            else:
                parent = root
                root = root.right
        if parent.val > val:
            parent.left = TreeNode(val)
        else:
            parent.right = TreeNode(val)
        return roott