[Solved] Given two binary trees original and cloned and given a reference to a node target in the original tree. The cloned tree is a copy of the original tree.

Question

Given two binary trees original and cloned and given a reference to a node target in the original tree.

The cloned tree is a copy of the original tree.

Return a reference to the same node in the cloned tree.

Note that you are not allowed to change any of the two trees or the target node and the answer must be a reference to a node in the cloned tree.

Example 1:

Machine Learning Projects
Input: tree = [7,4,3,null,null,6,19], target = 3
Output: 3
Explanation: In all examples the original and cloned trees are shown. The target node is a green node from the original tree. The answer is the yellow node from the cloned tree.

Example 2:

Machine Learning Projects
Input: tree = [7], target =  7
Output: 7

Example 3:

Machine Learning Projects
Input: tree = [8,null,6,null,5,null,4,null,3,null,2,null,1], target = 4
Output: 4

Constraints:

  • The number of nodes in the tree is in the range [1, 104].
  • The values of the nodes of the tree are unique.
  • target node is a node from the original tree and is not null.

Follow up: Could you solve the problem if repeated values on the tree are allowed?

Python Solution

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def getTargetCopy(self, original: TreeNode, cloned: TreeNode, target: TreeNode) -> TreeNode:
        q_org = []
        q_clo = []
        q_org.append(original)
        q_clo.append(cloned)
        if q_org[0]==target:
            return q_clo[0]

        while len(q_org)!=0:
            if q_org[0].left:
                if q_org[0].left==target:
                    return q_clo[0].left 
                q_org.append(q_org[0].left)
                q_clo.append(q_clo[0].left)
                
            if q_org[0].right:
                if q_org[0].right==target:
                    return q_clo[0].right
                q_org.append(q_org[0].right)
                q_clo.append(q_clo[0].right)
            q_org.pop(0)
            q_clo.pop(0)
Abhishek Sharma
Abhishek Sharma

Started my Data Science journey in my 2nd year of college and since then continuously into it because of the magical powers of ML and continuously doing projects in almost every domain of AI like ML, DL, CV, NLP.

Articles: 514

Leave a Reply

Your email address will not be published. Required fields are marked *