LeetCode——二叉树中的插入操作 Java

@author: sdubrz
@date: 2020.04.15
题目难度： 中等
考察内容： 树
原题链接 https://leetcode-cn.com/problems/insert-into-a-binary-search-tree/
题目的著作权归领扣网络所有，商业转载请联系官方授权，非商业转载请注明出处。
解题代码转载请联系 lwyz521604#163.com

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

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

例如,

给定二叉搜索树:

        4
       / \
      2   7
     / \
    1   3

和 插入的值: 5

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

         4
       /   \
      2     7
     / \   /
    1   3 5

或者这个树也是有效的:

         5
       /   \
      2     7
     / \   
    1   3
         \
          4

通过次数10,019 提交次数13,577

解法

这道题是一个二叉树的基本操作，虽然系统中给这道题的评级为中等，但其实是一个简单级别的。关于二叉树的基本操作在之前的博文二叉搜索树的Java实现中已经实现过。这里直接放出代码：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode insertIntoBST(TreeNode root, int val) {
        if(root==null){
            root = new TreeNode(val);
            return root;
        }

        TreeNode current = root;
        TreeNode p = root;
        while(current!=null){
            p = current;
            if(val>current.val){
                current = current.right;
            }
            else{
                current = current.left;
            }
        }

        if(val>p.val){
            p.right = new TreeNode(val);
        }else{
            p.left = new TreeNode(val);
        }

        return root;
    }
}

在 LeetCode 系统中提交轻松通过：

执行结果： 通过 显示详情
执行用时 : 0 ms, 在所有 Java 提交中击败了 100.00% 的用户
内存消耗 : 39.9 MB, 在所有 Java 提交中击败了 14.29% 的用户