Skip to main content

Function to add a node to binary search tree

Binary search tree (BST) is an ordered tree where each node has smaller values on the left sub tree and larger values on the right subtree. This is similar to binary search algorithm. And another useful fact is if you traverse the node in in-order method, the values of the tree will be in ascending order.


Binary Search Tree with root as 50

Now as you can see from the diagram on the left child and their children of root (which in this case is 50), values are 40, 20 and 45. All are less than 50. On the other hand, on the right sub tree 50, the values are 60, 55 and 78 - larger than 50.
And this rule follows for each and every node, not just root.

Now if we have to  add a node - let us 43, where shall we add it, in such a way that the tree will still remain as BST. The new node must be a leaf node (i.e. the node with no child nodes)

Let us traverse from root and find out where this new node fits. 43 is smaller than 50. Hence we must branch to left of 50. Now we get 40. 43 is larger than 40. Hence we must take a path to the right of 40. Here we come across 45. 43 is smaller than 45, so must take a path to the left. Luckily there are no nodes to the left of 45 (at last!). So we can attach our new node with 43, as left child of 45.

Similarly with the help of diagram, we can see that the nodes 59, 12,44 will be attached as right child of 55, left child of 20 and right child of 43 respectively.

Now how do we code all this?

We can either write a recursive function - which is easy to write but hard to understand, or we can write a non-recursive function. You must understand that, entire BST depends on recursive functions.

Anyways, for the sake of simplicity, let us first write a non-recursive function to add a node to BST.

  • for any given node
    • store node as prev_node
    •  if new value is > node value
      •  branch to right
    • if new value is < node value
      •  branch to left
    • if new value == node value
      • throw an error saying duplicate values are not allowed

 struct treenode
 { 
     int data; 
     struct treenode *left,*right;
 };
 tyepdef treenode *TREENODEPTR;

This is our structure for node of tree.  Next let us write a function createNode(). This function just allocates memory to a node and fills the data in it and sets left and right children to NULL. It returns the address of this newly created node.


TREENODEPTR createNode(int val) 
{ 
    TREENODEPTR newnode =  (TREENODEPTR)malloc(sizeof(struct treenode)); 
    newnode->data = val; newnode->left = NULL; 
    newnode->right = NULL; 
} 
 Here is the function to add this newnode to the tree TREENODEPTR. This function is non-recursive. 


void insertIntoBst(TREENODEPTR root, int newval)
{ 
 TREENODEPTR temp, prevnode,newnode = createNode(newval);
 /*we have an empty list. This node becomes root*/ 
if(root==NULL) 
 {
      root = newnode; 
      return root;
 } 
temp = root; 
while(1) 
{ 
    if (temp==NULL)
      { 
       ( newval > prevnode->data) ?(prevnode->right = newnode):(prevnode->left= newnode); 
       return root; 
      } 
    else 
    { 
         prevnode = temp; 
         if(newval > temp->data) 
            temp = temp->right;
         else if (newval < temp->data)
            temp = temp -> left; 
         else 
           { 
                  printf("Duplicate nodes are not allowed"); 
                 exit(1);
           }
     }
 } 
} 

But the function is so complicated. Because it should never be used. Any binary tree operation must be done using recursive function.

So let us write a recursive function to add a node to BST.

TREENODEPTR insertRecursive(TREENODEPTR node, int newval)
{
    TREENODEPTR newnode = createNode(newval);
    if(node==NULL)
       node = newnode;
    else if(newval>node->data)
       node->right = insertRec(node->right,newval);
    else if (newvaldata)
       node->left = insertRec(node->left,newval);
    else
       {
             printf("Duplicate nodes are not allowed");  
             exit(1);
    }
       
     return node;
} 

Isn't it elegant and short?


Comments

Popular posts from this blog

Delete a node from doubly linked list

Deletion operation in DLL is simpler when compared to SLL. Because we don't have to go in search of previous node of to-be-deleted node.  Here is how you delete a node Link previous node of node of to-be-deleted to next node. Link next node of node of to-be-deleted to previous node. Free the memory of node of to-be-deleted Simple, isn't it. The code can go like this. prevnode = delnode->prev; nextnode = delnode->next; prevnode->next = nextnode; nextnode->prev = prevnode; free(delnode); And that is it. The node delnode is deleted. But we should always consider boundary conditions. What happens if we are trying to delete the first node or last node? If first node is to be deleted, its previous node is NULL. Hence step 3 should not be used.  And also, once head is deleted, nextnode becomes head . Similarly if last node is to be deleted, nextnode is NULL. Hence step 4 is as strict NO NO. And we should set prevnode to tail. After we put these things together, we have...

Function to sort an array using bubble sort

Quick and dirty way of sorting an array is bubble sort. It is very easy to write and follow. But please keep in mind that it is not at all effecient. #include<iostream> using std::cin; using std::cout; void readArray(int arr[],int sz); void printArray(int arr[],int sz); void sortArray(int arr[],int sz); void swap(int &a,int &b); int main() {    int sz;    cout<<"Size of the array=";    cin>>sz;    int arr[sz];    readArray(arr,sz);     sortArray(arr,sz);   cout<<"Sorted array is ";   printArray(arr,sz); } void readArray(int arr[],int sz) {  for(int i=0;i<sz;i++)    {       cout<<"arr["<<i<<"]=";       cin>>arr[i];   } } void printArray(int arr[],int sz) {  for(int i=0;i<sz;i++)    {       cout<<"arr["<<i<<"]=";    ...

Merge two binary search trees

How do you merge two binary search trees? I googled about the solutions. Most solutions told me to convert both trees into linked lists. Merge the lists. Then create a tree from the elements of the list. But why lists? Why can't we store the elements in an array? Because if the data of the tree is larger - not just integer keys, array manipulation becomes difficult. But again, we need not convert both the trees into lists. We can convert one tree into list - a doubly linked list. Then insert the elements of this list into the other tree. I tried this approach. To convert a tree into a sorted doubly linked list Create a doubly linked list. Let the prev and next links of nodes in this list be called left and right respectively. This way we can directly use the binary tree nodes in the list. Use a static variable previousnode  call the function recursively for left child of current node. link current node to the previousnode set next pointer of previousnode to curre...