P53_2.kt

package ch14

import datatype.TreeNode
import kotlin.math.abs

class P53_2 {
    fun isBalanced(root: TreeNode?): Boolean {
        fun dfs(node: TreeNode?): Int {
            // 예외 처리
            if (node == null)
                return 0
            // 재귀 DFS로 왼쪽 리프 노드까지 탐색
            val left = dfs(node.left)
            // 재귀 DFS로 오른쪽 리프 노드까지 탐색
            val right = dfs(node.right)
            // 높이 균형이 아닌 경우 -1 리턴
            return if (left == -1 || right == -1 || abs(left - right) > 1)
                -1
            else
                // 왼쪽과 오른쪽 중 높은 노드 높이 +1 리턴
                left.coerceAtLeast(right) + 1
        }

        // 높이 균형이 깨진 경우가 아니라면 True, -1이 리턴되어 높이 균형이 깨진 경우 False 리턴
        return dfs(root) != -1
    }
}