给定二叉搜索树(BST)的根节点和要插入树中的值,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 保证原始二叉搜索树中不存在新值。
注意,可能存在多种有效的插入方式,只要树在插入后仍保持为二叉搜索树即可。 你可以返回任意有效的结果。
例如,
给定二叉搜索树: 4 / \ 2 7 / \ 1 3 和 插入的值: 5
你可以返回这个二叉搜索树:
4 / \ 2 7 / \ / 1 3 5
或者这个树也是有效的:
5 / \ 2 7 / \ 1 3 \ 4
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def insertIntoBST(self, root, val):
if not root: return TreeNode(val)#如果没有根就返回这个节点
temp = root
while 1:
if temp.val > val:#如果当前节点的值比val要大,那么val应该放在它的左边
if not temp.left:
temp.left = TreeNode(val)
break
else:
temp = temp.left
else:
if not temp.right:
temp.right = TreeNode(val)
break
else:
temp = temp.right
return root