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

给定二叉搜索树（BST）的根节点和要插入树中的值，将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 保证原始二叉搜索树中不存在新值。

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

例如,

给定二叉搜索树:

        4
       / \
      2   7
     / \
    1   3

和 插入的值: 5

你可以返回这个二叉搜索树:

         4
       /   \
      2     7
     / \   /
    1   3 5

或者这个树也是有效的:

         5
       /   \
      2     7
     / \   
    1   3
         \
          4

思路：

• 递归的方法
• 当根节点为空时，返回一个值为val的树节点
• 当val大于根节点值时，判断根节点是否有右子节点，如果没有，将值为val的树节点作为根节点的右子节点，否则在根节点的右子树中插入
• 当val小于根节点值时，判断根节点是否有左子节点，如果没有，将值为val的树节点作为根节点的左子节点，否则在根节点的左子树中插入

# 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):
        """
        :type root: TreeNode
        :type val: int
        :rtype: TreeNode
        """
        
        if not root:
            root = TreeNode(val)
        
        if val < root.val:
            if not root.left:
                root.left = TreeNode(val)
            else:
                root.left = self.insertIntoBST(root.left, val)
        
        else:
            if not root.right:
                root.right = TreeNode(val)
            else:
                root.right = self.insertIntoBST(root.right, val)
        
        return root

转载于:https://www.cnblogs.com/yuyin/articles/9675461.html