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

    Program to delete a node from linked list

    How do you remove a node from a linked list? If you have to delete a node, first you need to search the node. Then you should find its previous node. Then you should link the previous node to the next node. If node containing 8 has to be deleted, then n1 should be pointing to n2. Looks quite simple. Isn't it? But there are at least two special cases you have to consider. Of course, when the node is not found. If the node is first node of the list viz head. If the node to be deleted is head node, then if you delete, the list would be lost. You should avoid that and make the second node as the head node. So it becomes mandatory that you return the changed head node from the function.   Now let us have a look at the code. #include<stdio.h> #include<stdlib.h> struct node { int data; struct node * next; }; typedef struct node * NODEPTR; NODEPTR create_node ( int value) { NODEPTR temp = (NODEPTR) malloc( size...

    Binary tree deletion

    Do not get all scared and worried. It is not rocket science (or as I would like to call it - it is not regex). Remember deleting a node from linked list. When you deleted a node, you did the following 2 steps Free the memory of the node Link the previous node of the deleted node to the next node You will have to do these things in binary tree too. But the difficulty here is do you link the previous node - or parent node in tree terminology, to the left child of deleted node? Or to the right child? You can not link both children, as the parent already may have one child node. So before we solve this, let us categorize the deletion with the help of a diagram. Consider the cases The node to be deleted is a leaf node. That is, it does not have left or right child. In this case, link the parent to NULL in place of deleted node.  If you want to delete 13 - which is a leaf node, you set 14->left to NULL and free memory of node 13. If you want to delete 7 - which is also a leaf node and...

    Program to create a Linked List in C

    An array is a commonly used data structure in most of the languages. Because it is simple, it needs O(1) time for accessing elements. It is also compact. But an array has a serious drawback - it can not grow or shrink. You need to estimate the array size and define it during compile time. This drawback is not present a linked list. A linked list is a data structure which can grow or shrink dynamically.  A linked list has nodes each of which contain  contain  data and a link to next node . These nodes are dynamically allocated structures. If you need more nodes, you just need to allocate memory for these and link these nodes to the existing list. The nodes of a linked list have to be defined as self-referential structures in C. That is structures with data members and one member which is a pointer to the structure of same type.  This pointer will work as a link to next node. struct node { int data; struct node * next; //pointer to another node }...