리트코드 226. Invert Binary Tree

27 Nov 2022 in Algorithm

이진 트리를 좌우로 뒤집어보자!

Tree 문제… 🎋

문제

226. Invert Binary Tree 문제 링크

이진 트리를 좌우 반대로 만들어 반환하시오!

# Example 1:
Input: root = [4,2,7,1,3,6,9]
Output: [4,7,2,9,6,3,1]

예제를 보시면 각 노드의 왼쪽, 오른쪽 하위 노드들이 반대로 바뀐 것을 볼 수 있습니다.

✅ 최종 Solution

이건 간단하게 재귀로 반복하면서 좌우로만 바꾸면 됩니다.
노드가 2개 뿐이라 굉장히 간단하게 풀 수 있었습니다. 😁

def invertTree(self, root: Optional[TreeNode]) -> Optional[TreeNode]:
  def invert(node):
    if not node: return
    node.left, node.right = node.right, node.left
    invert(node.left)
    invert(node.right)
  invert(root)
  return root