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

Linked list in C++

A linked list is a versatile data structure. In this structure, values are linked to one another with the help of addresses. I have written in an earlier post about how to create a linked list in C.  C++ has a library - standard template library which has list, stack, queue etc. data structures. But if you were to implement these data structures yourself in C++, how will you implement? If you just use new, delete, cout and cin, and then claim it is your c++ program, you are not conforming to OOPS concept. Remember you have to "keep it together". Keep all the functions and variables together - in a class. You have to have class called linked list in which there are methods - append, delete, display, insert, find, find_last. And there will also be a data - head. Defining node We need a structure for all these nodes. A struct can be used for this purpose, just like C. struct node { int val; struct node * next; }; Next we need to define our class. W

Swap nodes of a linked list

Qn: Write a function to swap the adjacent nodes of a singly linked list.i.e. If the list has nodes as 1,2,3,4,5,6,7,8, after swapping, the list should be 2,1,4,3,6,5,8,7 Image from: https://tekmarathon.com Though the question looks simple enough, it is tricky because you don't just swap the pointers. You need to take care of links as well. So let us try to understand how to go about it. Take two adjacent nodes p1 and p2 Let prevnode be previous node of p1 Now link prevnode to p2 Link p2 to p1 Link p1 to next node of p2 So the code will be prevnode -> next = p2; p1 -> next = p2 -> next; p2 -> next = p1; But what about the start node or head? head node does not have previous node If we swap head with second node, modified head should be sent back to caller  To take care of swapping first and second nodes, we can write p1 = head; p2 = head -> next; p1 -> next = p2 -> next; p2 -> next = p1; head = p2;  Now we are read

Binary tree deletion - non-recursive

In the previous post we have seen how to delete a node of a binary search tree using recursion. Today we will see how to delete a node of BST using a non-recursive function. Let us revisit the 3 scenarios here Deleting a node with no children just link the parent to NULL Deleting a node with one child link the parent to  non-null child of node to be deleted Deleting a node with both children select the successor of node to be deleted copy successor's value into this node delete the successor In order to start, we need a function to search for a node in binary search tree. Did you know that searching in  a BST is very fast, and is of the order O(logn). To search Start with root Repeat until value is found or node is NULL If the search value is greater than node branch to right If the search value is lesser than node branch to left.  Here is the function NODEPTR find_node (NODEPTR root,NODEPTR * parent, int delval) { NODEPTR nd = root; NODEPTR pa = root; if (root -> v