package ch14;

import datatype.TreeNode;

public class P53_1 {
    public int dfs(TreeNode node) {
        // 예외 처리
        if (node == null)
            return 0;
        // 재귀 DFS로 왼쪽 리프 노드까지 탐색
        int left = dfs(node.left);
        // 재귀 DFS로 오른쪽 리프 노드까지 탐색
        int right = dfs(node.right);
        // 높이 균형이 아닌 경우 -1 리턴
        if (left == -1 || right == -1 || Math.abs(left - right) > 1)
            return -1;
        // 왼쪽과 오른쪽 중 높은 노드 높이 +1 리턴
        return Math.max(left, right) + 1;
    }

    public boolean isBalanced(TreeNode root) {
        // 높이 균형이 깨진 경우가 아니라면 true, -1이 리턴되어 높이 균형이 깨진 경우 false 리턴
        return dfs(root) != -1;
    }
}