[Python] 226. 트리 데칼코마니

leetcode.com/problems/invert-binary-tree/

Invert Binary Tree - LeetCode

바이너리 트리를 데칼코마니 시키는 것을 말한다.

좌측과 우측을 바꾸는 문제

 

재귀적으로 생각해서 현재 트리의 좌측과 우측을 구한다음 그것을 현재 트리의 좌측-우측, 우측-좌측 이렇게 매핑 시켜주면 된다

class Solution:
    def invertTree(self, root: TreeNode) -> TreeNode:
        if not root: return None
        right = self.invertTree(root.right)
        left = self.invertTree(root.left)
        root.left = right
        root.right = left
        return root

 

구현한 것만 보면 쉬운데 재귀적으로 생각하기가 어려운 것 같다.