[leetcode 226] Invert Binary Tree

226. Invert Binary Tree

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

해결 과정

왼쪽 트리랑 오른쪽 트리를 swqp 하는 문제로 이해했다. 트리 아래로 내려가며 leaf가 null 일때까지 반복하면서 (탈출조건) swap 시키는 코드를 작성했다.

코드

fun invertTree(root: TreeNode?): TreeNode? {
  if(root == null) return null
        
  val left = invertTree(root.left)
  val right = invertTree(root.right)
  
  root.left = right
  root.right = left
  return root
}

배운 점

leftright 에 left와 right 값을 메모리에 들고 있기 때문에

root.left = right 로 swqp을 할 때 temp 값이 필요 없다.

Leave a comment