Skip to main content

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
    1. 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.
    2. Use a static variable previousnode 
    3. call the function recursively for left child of current node.
    4. link current node to the previousnode
      1. set next pointer of previousnode to current node. 
      2. set previous pointer of current node to previous node
    5. Now set this node as previousnode
    6. call the function for right child of current node. 


    If previousnode is NULL, it means you are visiting tree minimum, so set this node as head.

    Here is the function to convert binary search tree to sorted doubly linked list. 

    void convert_to_dll(NODEPTR nd,NODEPTR*headptr)
    {
       static NODEPTR prevnode;
       if(nd!=NULL)
        {       
     convert_to_dll(nd->left,headptr);
            if(prevnode!=NULL){
               prevnode->right = nd;
               nd->left = prevnode;
            }
            else{
               *headptr = nd;
            }
            prevnode = nd; 
            convert_to_dll(nd->right,headptr);
         }
    }
    
    Next we need to take this list, extract one node at a time and insert the node in the other binary search tree.  
    How do we extract nodes? Easiest way to do that is take head node (first node), insert this to BST. Move to next node in the list by saying head = head->next.
    Please remember that, we need to set both links of the node to NULL before inserting it into BST.
    1. set temp to head
    2. insert temp into binary search tree
    3. move to next node by using head = head->right. (right because we have used left for previous and right for next)
    4. repeat the steps 1 to 3  as long as  head is not NULL. 
    Here is the code for the same.


    NODEPTR merge_bsts(NODEPTR root,NODEPTR head)
    {
         while(head!=NULL)
         {
     NODEPTR temp = head;
     head = head->right;
            temp->left = temp->right = NULL;
     root = insert_node(root,temp);
          }
          return root;
    }
    

    Let us write a create 2 lists and use these functions.


    #include<stdio.h>
    #include<stdlib.h>
    struct node
    {
       int val;
       struct node *left;
       struct node *right;
    };
    typedef struct node *NODEPTR;
    
    NODEPTR create_node(int num)
    {
         NODEPTR temp = (NODEPTR)malloc(sizeof(struct node));
         temp->val = num;
         temp->left = NULL;
         temp->right = NULL;
         return temp;
    } 
    
    NODEPTR insert_node(NODEPTR nd,NODEPTR newnode)
    {
        if(nd==NULL)
           return newnode;/* newnode becomes root of tree*/
        if(newnode->val > nd->val)
            nd->right = insert_node(nd->right,newnode);
        else if(newnode->val <  nd->val)
            nd->left = insert_node(nd->left,newnode); 
        return nd;   
    }
    
    void convert_to_dll(NODEPTR nd,NODEPTR*headptr)
    {
       static NODEPTR prevnode;
       if(nd!=NULL)
        {       
     convert_to_dll(nd->left,headptr);
            if(prevnode!=NULL){
               prevnode->right = nd;
               nd->left = prevnode;
            }
            else{
               *headptr = nd;
            }
            prevnode = nd; 
            convert_to_dll(nd->right,headptr);
         }
    }
     
    void print_dll(NODEPTR nd)
    {
       NODEPTR temp = nd;
        while(nd)
          {
            printf("%d---->",nd->val);
            temp = nd; nd = nd->right;
           }
         printf("\nnow in reverse");
         nd = temp;
         while(nd)
          {
            printf("%d---->",nd->val);
            temp = nd; nd = nd->left;
           }
    }
    
    NODEPTR merge_bsts(NODEPTR root,NODEPTR head)
    {
         while(head!=NULL)
         {
     NODEPTR temp = head;
     head = head->right;
            temp->left = temp->right = NULL;
     root = insert_node(root,temp);
          }
          return root;
    }
    
    void inorder(NODEPTR nd)
    {
       if(nd!=NULL)
       { 
     inorder(nd->left);
     printf("%d ",nd->val);
            inorder(nd->right);
        }
    }
    
    int main()
    {
           NODEPTR root=NULL,delnode,head; 
           NODEPTR root2 = NULL;
           int n;
           do
           {
               NODEPTR newnode;
               printf("Enter value of node(-1 to exit):");
               scanf("%d",&n);
               if(n!=-1)
                {  
                   newnode = create_node(n);
                   root = insert_node(root,newnode);
                 }
           } while (n!=-1);
           printf("Second bst:");
           do
           {
               NODEPTR newnode;
               printf("Enter value of node(-1 to exit):");
               scanf("%d",&n);
               if(n!=-1)
                {  
                   newnode = create_node(n);
                   root2 = insert_node(root2,newnode);
                 }
           } while (n!=-1);
           head = NULL;
           convert_to_dll(root2,&head);
           print_dll(head);
           root  = merge_bsts(root,head); 
           inorder(root);
           return 0;
    }
    

    Comments

    Popular posts from this blog

    Introduction to AVL tree

    AVL tree is a balanced binary search tree where the difference between heights of two sub trees is maximum 1. Why balanced tree A binary tree is good data structure because search operation here is of the order of O(logn). But this is true if the tree is balanced - which means the left and right subtrees are almost equal in height. If not balanced, search operation will take longer.  In worst case, if the tree has only one branch, then search is of the order O(n). Look at this example.  Here all nodes have only right children.  To search a value in this tree, we need upto 7 iterations, which is O(n). So this tree is very very inefficient. One way of making the tree efficient is to, balance the tree and make sure that height of two branches of each node are almost equal. Height of a node Height of a node is the distance between the node and its extreme child.  In the above a diagram, height of 37 is 3 and height of left child of 37 is 0 and right child of 37 is 2. Bal...

    Reverse a singly linked list

    One of the commonly used interview question is - how do you reverse a linked list? If you talk about a recursive function to print the list in reverse order, you are so wrong. The question is to reverse the nodes of list. Not print the nodes in reverse order. So how do you go about reversing the nodes. You need to take each node and link it to previous node. But a singly linked list does not have previous pointer. So if n1 is current node, n2 = n1->next, you should set     n2->next = NULL But doing this would cut off the list at n2. So the solution is recursion. That is to reverse n nodes  n1,n2,n3... of a list, reverse the sub list from n2,n3,n4.... link n2->next to n1 set n1->next to NULL The last step is necessary because, once we reverse the list, first node must become last node and should be pointing to NULL. But now the difficulty is regarding the head? Where is head and how do we set it? Once we reach end of list  viz n1->next ==NULL, th...

    Balanced brackets

    Have you observed something? When ever you are writing code using any IDE, if you write mismatched brackets, immediately an error is shown by IDE. So how does IDE  know if an expression is having balanced brackets? For that, the expression must have equal number of opening and closing brackets of matching types and also they must be in correct order. Let us look at some examples (a+b)*c+d*{e+(f*g)}   - balanced (p+q*[r+u )] - unbalanced (p+q+r+s) ) - unbalanced (m+n*[p+q]+{g+h}) - balanced So we do we write a program to check if an expression is having balanced brackets? We do need to make use of stack to store the brackets. The algorithm is as follows Scan a character - ch from the expression If the character is opening bracket, push it to stack If the character is closing bracket pop a character from stack If popped opening bracket and ch are not of same type ( ( and ) or [ and ] ) stop the function and return false Repeat steps 2 and 3 till all characters are scanned. On...