[LintCode] Insert Node in a Binary Search Tree

Given a binary search tree and a new tree node, insert the node into the tree. You should keep the tree still be a valid binary search tree.

You can assume there is no duplicate values in this tree + node.

Example

Given binary search tree as follow, after Insert node 6, the tree should be:

  2             2
 / \           / \
1   4   -->   1   4
   /             / \ 
  3             3   6

Challenge 

Can you do it without recursion?

 

Solution 1. Recursion

Recursively find the correct subtree of the current processing node to do the insertion. If current node's value is bigger than the insert node, go to its left subtree; otherwise go to its right subtree. 

Each recursive call returns the new root node after the insertion node is inserted.

 

/**
 * Definition of TreeNode:
 * public class TreeNode {
 *     public int val;
 *     public TreeNode left, right;
 *     public TreeNode(int val) {
 *         this.val = val;
 *         this.left = this.right = null;
 *     }
 * }
 */
public class Solution {
    public TreeNode insertNode(TreeNode root, TreeNode node) {
        if(root == null){
            return node;
        }
        if(node.val < root.val){
            root.left = insertNode(root.left, node);
        }
        else{
            root.right = insertNode(root.right, node);
        }
        return root;
    }
}

 

Solution 2. Iterative approach 

The core idea of inserting a node in BST is to find the "right" place to insert.  In solution 1, this place is found by recursively calling the insertNode method. 

When the current node is null, it means that the insertion place is just found, return the insertion node as the root node of this null subtree.  Each recursive call

returns the new node after inserting node to the subtree. 

 

Alternatively, we can found this insertion place iteratively.  The only difference is that we have to do the info booking by ourselves now.   Inserting a node essentially

means we need to set a parent node's left or right node to the insertion node. As a result, we need to save the parent node of the current processing node. When 

the current processing node is null, we found the "right" insertion place. Then we need to check if the node should be inserted to its parent's left or right. 

public class Solution {
    public TreeNode insertNode(TreeNode root, TreeNode node) {
        if(root == null){
            return node;
        }    
        TreeNode currNode = root;
        TreeNode parent = null;
        while(currNode != null){
            parent = currNode;
            if(node.val < currNode.val){
                currNode = currNode.left;
            }
            else{
                currNode = currNode.right;
            }
        }
        if(node.val < parent.val){
            parent.left = node;
        }
        else{
            parent.right = node;
        }
        return root;
    }
}

 

 

Related Problems 

Remove Node in Binary Search Tree