DataScience trainee

Leetcode 1379. Find a Corresponding Node of a Binary Tree in a Clone of That Tree in Python3

Find a Corresponding Node of a Binary Tree in a Clone of That Tree


개요

  • 이번 포스팅은 파이썬 코딩테스트 연습문제 풀이입니다.
  • Leetcode 에 공개된 예제이며 링크는 아래와 같습니다.

https://leetcode.com/problems/find-a-corresponding-node-of-a-binary-tree-in-a-clone-of-that-tree/


코드

def getTargetCopy(self, original: TreeNode, cloned: TreeNode, target: TreeNode) -> TreeNode:
    from collections import deque

    queue = deque()
    queue.append(cloned)

    while queue:
        current = queue.popleft()

        if current.val == target.val:
            return current

        if current.left != None:
            queue.append(current.left)
        if current.right != None:
            queue.append(current.right)

마치며

이번 포스팅은 Leetcode에 존재하는DFS 태크 중 난이도 Medium 에 해당하는 문제입니다.
저는 BFS로 풀었지만 DFS로도 풀이가 가능하기 때문에 태그가 중복된 것 같습니다.
추가적으로 저는 고려하지 않았지만 트리 내에 val 값의 중복이 허용을 고려하고 문제를 푼다면 더욱 난이도가 올라 갈 것 입니다.