2019-10-14-LeetCode_110_Balanced_Binary_Tree.markdown

layout	title	description	date	categories	comments
post	[LeetCode 110] Balanced Binary Tree	[LeetCode 110] Balanced Binary Tree	2019-10-14 11:30:00 -0700	Algorithm	true

[LeetCode 110] Balanced Binary Tree

문제 요약

이진트리가 주어졌을 때, 균형잡힌 이진트리인지 판별하는 문제

풀이

재귀함수를 활용하여 트리를 순회하며 왼쪽 서브트리의 깊이와 오른쪽 서브트리의 깊이 차이가 2 이상 나는 것을 확인해준다.

코드

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    private boolean ret = true;
    
    private int go(TreeNode tn, int d) {
        if (tn == null) return d;
        int ld = go(tn.left, d + 1), rd = go(tn.right, d + 1);
        if (Math.abs(ld - rd) > 1) ret = false;
        return Math.max(ld, rd);
    }
    
    public boolean isBalanced(TreeNode root) {
        go(root, 0);
        return ret;
    }
}