Skip to main content

Check if two binary trees are equal

Question: Write a function to test if two binary trees are equal. The trees are equal only if the values are equal and also the structures are equal.


Equal trees

      89                                89
  12      100                    12       100
2                                 2

Not equal trees
        89                                   89
   2        100                      12        100
     12                             2

First thing which comes to our mind is just write

if   nd1->val !=nd2->val, trees are not equal.

OK. That is correct. But then what? What do we compare next? Value of left child? Or value of right child?

Like all other operations of binary tree, this function also needs recursion.

After ensuring that nd1->val and nd2->val are equal, we have to call this function on left child and right child. And only if both of these calls return 1, the trees are equal.

So here are the steps
  • If nodes of two trees nd1 and nd2  are not null
    • If values of nodes are not equal return false
    • If not, 
      • l = return value of function called for left children of nd1 and nd2
      • r = return value of function for right children of nd1 and nd2
      • return l && r
  • Now if both are NULL return true
 If both nd1 and nd2 are equal, then they are leaf nodes and hence we must return true. As long as both are having same values in subtrees, l&&r will be 1 (which is true in C).

But if there is a non-matching node somewhere, then the function returns false, which is propagated up the stack.

Let us write the function.


int equal_trees(NODEPTR nd1,NODEPTR nd2)
{
    if(nd1==nd2)
       return 1;//both are null
    if(nd1!=NULL ){
      if(nd2==NULL)
         return 0;
      if(nd1->val!=nd2->val)
         return 0;
      else
         {
           int l= equal_trees(nd1->left,nd2->left);
           int r = equal_trees(nd1->right,nd2->right);
           return l&r;
         }
    }
    if(nd2!=NULL && nd1==NULL)
       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...