码迷,mamicode.com
首页 > 其他好文 > 详细

LeetCode 701. Insert into a Binary Search Tree

时间:2019-06-13 14:00:56      阅读:123      评论:0      收藏:0      [点我收藏+]

标签:into   turn   ali   ret   otherwise   space   style   may   multi   

原题链接在这里:https://leetcode.com/problems/insert-into-a-binary-search-tree/

题目:

Given the root node of a binary search tree (BST) and a value to be inserted into the tree, insert the value into the BST. Return the root node of the BST after the insertion. It is guaranteed that the new value does not exist in the original BST.

Note that there may exist multiple valid ways for the insertion, as long as the tree remains a BST after insertion. You can return any of them.

For example, 

Given the tree:
        4
       /       2   7
     /     1   3
And the value to insert: 5

You can return this binary search tree:

         4
       /         2     7
     / \   /
    1   3 5

This tree is also valid:

         5
       /         2     7
     / \   
    1   3
                   4

题解:

If root.val < val, then must insert on root right side, if root right side is already null, then root.right = new TreeNode(val). Otherwise, keep iterating on root.right.

Vice versa.

Time Complexity: O(h). h is height of tree.

Space: O(1).

AC 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){
            return new TreeNode(val);
        }
        
        TreeNode cur = root;
        while(cur != null){
            if(cur.val < val){
                if(cur.right == null){
                    cur.right = new TreeNode(val);
                    return root;
                }
                
                cur = cur.right;
            }else{
                if(cur.left == null){
                    cur.left = new TreeNode(val);
                    return root;
                }
                
                cur = cur.left;
            }
        }
        
        return null;
    }
}

 

LeetCode 701. Insert into a Binary Search Tree

标签:into   turn   ali   ret   otherwise   space   style   may   multi   

原文地址:https://www.cnblogs.com/Dylan-Java-NYC/p/11015910.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!