LeetCode 筆記 - 1379. 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 search(root):
            if not root:
                return None
            if root.val == target.val:
                return root
            
            if result := search(root.left):
                return result
            if result := search(root.right):
                return result
            
            return None
        
        if not original:
            return None
        
        return search(cloned)