Leetcode每日打卡20200930----二叉搜索树中的插入操作

二叉搜索树中的插入操作

给定二叉搜索树（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-108 <= val <= 10^8
• 新值和原始二叉搜索树中的任意节点值都不同
• 通过次数39,049提交次数54,337

• 思路：二叉搜索树： 对于树上的每个节点来说，该节点的左子树里所有的节点都小于当前节点，该节点的右子树里所有的节点都大于当前节点。这道题可以有递归和迭代两种写法。先来看递归

//递归
如果 root 是空，则新建树节点作为根节点返回即可。
否则比较 root.val 与目标值的大小关系：
如果 root.val 大于目标值，说明目标值应当插入 root 的左子树中，递归调用；
如果 root.val 小于目标值，说明目标值应当插入 root 的右子树中，递归调用。
//迭代思想和递归差不多

• 代码实现：

class Solution {
     public TreeNode insertIntoBST(TreeNode root, int val) {
        if(root == null){
            return new TreeNode(val);
        }
        if(root.val < val){
            root.right = insertIntoBST(root.right,val);
        } else {
            root.left = insertIntoBST(root.left,val);
        }
        return root;
    }
}

//迭代
public TreeNode insertIntoBST(TreeNode root, int val) {
        if(root == null){
            return new TreeNode(val);
        }
        TreeNode parent = root,p = root;
        while(p != null){
            parent = p;
            p = p.val < val ? p.right : p.left;
        }
        if(parent.val < val){
            parent.right = new TreeNode(val);
        } else {
            parent.left = new TreeNode(val);
        }
        return root;
    }