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

举报
悲恋花丶无心之人 发表于 2021/02/02 23:54:51 2021/02/02
【摘要】 目录 一、题目内容 二、解题思路 三、代码 一、题目内容 给定二叉搜索树(BST)的根节点和要插入树中的值,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据保证,新值和原始二叉搜索树中的任意节点值都不同。 注意,可能存在多种有效的插入方式,只要树在插入后仍保持为二叉搜索树即可。 你可以返回任意有效的结果。 例如,  ...

目录

一、题目内容

二、解题思路

三、代码


一、题目内容

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

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

例如, 

给定二叉搜索树:

        4
       / \
      2   7
     / \
    1   3

和 插入的值: 5
你可以返回这个二叉搜索树:

         4
       /   \
      2     7
     / \   /
    1   3 5
或者这个树也是有效的:

         5
       /   \
      2     7
     / \   
    1   3
         \
          4

 

提示:

给定的树上的节点数介于 0 和 10^4 之间
每个节点都有一个唯一整数值,取值范围从 0 到 10^8
-10^8 <= val <= 10^8
新值和原始二叉搜索树中的任意节点值都不同

二、解题思路

根据BST的性质,左小右大,因此递归的判断,直到对应子节点为空就插入。

三、代码


   
  1. # Definition for a binary tree node.
  2. class TreeNode:
  3. def __init__(self, val=0, left=None, right=None):
  4. self.val = val
  5. self.left = left
  6. self.right = right
  7. def __repr__(self):
  8. return str(self.val)
  9. class Solution:
  10. def insertIntoBST(self, root: TreeNode, val: int) -> TreeNode:
  11. if not root:
  12. return TreeNode(val)
  13. if root.val > val:
  14. root.left = self.insertIntoBST(root.left, val)
  15. else:
  16. root.right = self.insertIntoBST(root.right, val)
  17. return root
  18. if __name__ == '__main__':
  19. a = TreeNode(4)
  20. a.left = TreeNode(2)
  21. a.right = TreeNode(7)
  22. a.left.left = TreeNode(1)
  23. a.left.right = TreeNode(3)
  24. s = Solution()
  25. val = 5
  26. ans = s.insertIntoBST(a, val)
  27. print(ans)

文章来源: nickhuang1996.blog.csdn.net,作者:悲恋花丶无心之人,版权归原作者所有,如需转载,请联系作者。

原文链接:nickhuang1996.blog.csdn.net/article/details/108882660

【版权声明】本文为华为云社区用户转载文章,如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。