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% 的用戶