Skip to main content

Inorder successor of binary search tree

Question : Write a function to find inorder successor of a given node in a binary  search tree.

Our inclination would be to write tree minimum of right subtree. Because we already use this mechanism while deleting a node with both subtrees.

But what we fail to notice is that tree minimum of right subtree would be present if there IS a right subtree. What if the node has no right child? What if it is a leaf node? Which nodes are successors of these nodes? And how do we find them?

The inorder successor of 3 can be found out as tree minimum of right subtree, which is 4. But what about successors of 4? Or 7? Or successor 13?

One method would be to store parent link in each node. And travel up finding parent nodes until the parent node has a value greater than given node.

But that needs modification of the structure. Another method is to start from root node and search for a given value and store node with larger value as successor and terminate when we come across a node with value matching given node.

e.g. In the tree given above

    To find successor of 4, we start with root - successor = 8
        Then we branch left and visit node 3 - successor is still 8
        Then we branch right and visit node 6 - successor is 6
        Then we branch left and we terminate loop

     We found our successor as 4

To find successor of 7,
         We start with root  - successor = 8
         We branch to left - successor =8
         We branch to right - successor = 8
         We branch to right - we found 7 and terminate loop
         Successor is 8

Here is the function which covers both the cases i.e. the case where given node has a right child and the case where there is no right child and hence above mechanism is to be used.


NODEPTR successor(NODEPTR root, NODEPTR nd)
{
     if(nd->right!=NULL)/*has right child*/
     {
         nd = nd->right;
         while(nd->left!=NULL)
         { 
             nd = nd->left;
         }
         return nd;
     }
     else
     {
        NODEPTR succ = NULL;
        while(root!=NULL)
        {
            if(root->val > nd->val)
            {
         succ = root;
         root = root ->left;
            }
            else if(root->val <nd->val)
                root = root ->right;
            else
                return succ;
        }
        return root;
    } 
}

Also you have to remember the fact that, the right most node - the largest value in a BST has no successor.

Here is complete program

#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 in_order(NODEPTR nd)
{
   if(nd!=NULL)
    {
        in_order(nd->left);
        printf("%d---",nd->val);
        in_order(nd->right);
     }
}
NODEPTR search(NODEPTR nd,int num)
{
    while(nd!=NULL )
    {
        if (num > nd->val)
           nd = nd->right;
        else if (num< nd->val)
           nd = nd->left;
        else 
           return nd;
     }
     return NULL;/*not found*/
}

NODEPTR successor(NODEPTR root, NODEPTR nd)
{
     if(nd->right!=NULL)
     {
         nd = nd->right;
         while(nd->left!=NULL)
         { 
             nd = nd->left;
         }
         return nd;
     }
     else
     {
        NODEPTR succ = NULL;
        while(root!=NULL)
        {
            if(root->val > nd->val)
            {
  succ = root;
         root = root ->left;
            }
            else if(root->val <nd->val)
                root = root ->right;
            else
                return succ;
        }
        return root;
    } 
}


int main()
{
       NODEPTR root=NULL,delnode; 
       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("\nInorder traversal\n");
       in_order(root);  
       while(1)
       {   
         printf("Enter node value (-1 to stop):");
         scanf("%d",&n);
         if(n==-1)
            break;
         NODEPTR nd = search(root,n);
         if(nd==NULL)
             printf("%d is not present in tree",n);
         else
            {
             NODEPTR nds = successor(root,nd);
             if(nds!=NULL)
                printf("the successor is %d\n",nds->val);
             else
                printf("No successor\n");
            }
       }    
       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...

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...

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...