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

Delete a node from doubly linked list

Deletion operation in DLL is simpler when compared to SLL. Because we don't have to go in search of previous node of to-be-deleted node.  Here is how you delete a node Link previous node of node of to-be-deleted to next node. Link next node of node of to-be-deleted to previous node. Free the memory of node of to-be-deleted Simple, isn't it. The code can go like this. prevnode = delnode->prev; nextnode = delnode->next; prevnode->next = nextnode; nextnode->prev = prevnode; free(delnode); And that is it. The node delnode is deleted. But we should always consider boundary conditions. What happens if we are trying to delete the first node or last node? If first node is to be deleted, its previous node is NULL. Hence step 3 should not be used.  And also, once head is deleted, nextnode becomes head . Similarly if last node is to be deleted, nextnode is NULL. Hence step 4 is as strict NO NO. And we should set prevnode to tail. After we put these things together, we have...

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

Function to sort an array using bubble sort

Quick and dirty way of sorting an array is bubble sort. It is very easy to write and follow. But please keep in mind that it is not at all effecient. #include<iostream> using std::cin; using std::cout; void readArray(int arr[],int sz); void printArray(int arr[],int sz); void sortArray(int arr[],int sz); void swap(int &a,int &b); int main() {    int sz;    cout<<"Size of the array=";    cin>>sz;    int arr[sz];    readArray(arr,sz);     sortArray(arr,sz);   cout<<"Sorted array is ";   printArray(arr,sz); } void readArray(int arr[],int sz) {  for(int i=0;i<sz;i++)    {       cout<<"arr["<<i<<"]=";       cin>>arr[i];   } } void printArray(int arr[],int sz) {  for(int i=0;i<sz;i++)    {       cout<<"arr["<<i<<"]=";    ...