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

    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