1379_find-a-corresponding-node-of-a-binary-tree-in-a-clone-of-that-tree.py

"""
https://leetcode.com/problems/find-a-corresponding-node-of-a-binary-tree-in-a-clone-of-that-tree/
"""
# 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:
#         def traverse(og, copy):
#             if og:
#                 if og == target:
#                     self.answer = copy
                
#                 traverse(og.left, copy.left)
#                 traverse(og.right, copy.right)
        
#         traverse(original, cloned)
#         return self.answer

    def getTargetCopy(self, original: TreeNode, cloned: TreeNode, target: TreeNode) -> TreeNode:
        if not original:
            return
        
        if original == target:
            return cloned
        
        left = self.getTargetCopy(original.left, cloned.left, target)
        right = self.getTargetCopy(original.right, cloned.right, target)
        
        return left or right