[leetcode 226] Invert Binary Tree
226. 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
}
배운 점
left
와 right
에 left와 right 값을 메모리에 들고 있기 때문에
root.left = right
로 swqp을 할 때 temp 값이 필요 없다.
Leave a comment