Skip to main content

Merge Two Binary Trees



Given two binary trees and imagine that when you put one of them to cover the other, some nodes of the two trees are overlapped while the others are not.You need to merge them into a new binary tree. The merge rule is that if two nodes overlap, then sum node values up as the new value of the merged node. Otherwise, the NOT null node will be used as the node of new tree.

Example 1:

Input: 
	Tree 1                     Tree 2                  
          1                         2                             
         / \                       / \                            
        3   2                     1   3                        
       /                           \   \                      
      5                             4   7 
 
Output: 
Merged tree:
	     3
	    / \
	   4   5
	  / \   \ 
	 5   4   7
 
Note: The merging process must start from the root nodes of both trees.

 Source:LeetCode

Solution:

We traverse both trees in a preorder fashion. In each step, we check if both nodes isn't null, if so we add the value of the nodes; otherwise, we return the one that isn't null as a child of the first tree (t1). if in any case, it happens that the second tree (t2) has  reached leaf node then we return t1, and vice versa. At the end, the first tree (t1) will be the resultant tree which represent the merged tree.

Note: I haven't created  a new tree, I'm using the first tree to represent the final tree. However, doing so will follow similar approach as one shown below.

Complexity Analysis
  • Time complexity : O(m)O(m). mm nodes need to be traversed (mm represents the minimum number of nodes from the two given trees)
  • Space complexity : O(m)O(m). In average case, depth will be O(log(m)).O(logm).

 Recursive Solution:


 public TreeNode mergeTrees(TreeNode t1, TreeNode t2) {
     
        if(t1!=null && t2!=null)
            t1.val+= t2.val; //add the node's value
        else 
            return t1==null?t2:t1; //return appropriate node       
        
        t1.left =  mergeTrees(t1==null?null:t1.left,t2==null?null:t2.left);
        t1.right = mergeTrees(t1==null?null:t1.right,t2==null?null:t2.right);
        
        return t1;
    }

Comments

Post a Comment

Popular posts from this blog

RLE Encoding and Decoding in C++

Given an input string, write a regular recursive function that returns the decoded (uncompresseed form) Run Length Encoded   (is a simple form of data compression where repeated character are replaced by count followed by the character repeated) string for the input string. Below are some examples: decode("*") =>* decode("3+") =>+++ decode("11*") =>*********** decode("101+") =>+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ decode("abcde10+10*10+10x") =>abcde++++++++++**********++++++++++xxxxxxxxxx decode("\\") =>\ decode("\1\2\3") =>123 decode("13\7x") =>7777777777777x decode("5\\") =>\\\\\ decode("4\12\23\3") =>111122333 decode("4\\2\3") =>\\\\33 NOTE: To represent a single backslash, it’s necessary to place double backslashes (\\) in the input string to obtain the desired in

Invert Binary Search Tree

Invert a binary tree. Example: 4 / \ 2 7 / \ / \ 1 3 6 9   to   4 / \ 7 2 / \ / \ 9 6 3 1  4 / \ 2 7 / \ / \ 1 3 6 9   Trees are crucial data structure in Computer Science. Often trees related problems are solved using recursion as it provides an elegant solution in this particular problem at least. The basic terminology to solve this problem is to swap the left and right values, if you work on the tree from the bottom the to the top swapping the left and right node, the tree will be inverted. Let's go through an example: You can start from left or right. So, starting from the left we have   2 2 (swap the left and right node) / \ => / \ 1 3 3 1   7 7 (similarly, on the right sub-tree) / \ => / \ 6 9 9 6    Which gives,   4 / \ 2 7 / \ / \ 3 1 9 6 And finally, 4 (swap ag