검지offer 시리즈의 38: 균형 이차수 여부 판단

제목 설명
두 갈래 나무를 입력하여 이 두 갈래 나무가 균형 있는 두 갈래 나무인지 아닌지를 판단하세요.
이른바 균형 두 갈래 나무는 어떤 결점에 대해 말하자면 왼쪽 나무의 깊이와 오른쪽 나무의 깊이의 절대치는 1을 넘지 않는다.각 노드에 대한 판단이 필요하기 때문에 귀착적인 사고방식으로 해결할 수 있다.구체적인 사고방식은 먼저 뿌리 노드의 좌우 나무의 깊이를 구하고 이를 판단하는 것이다. 만약에 좌우 나무의 깊이의 절대치가 1을 넘지 않는 조건을 만족시키지 못하면 균형 두 갈래 나무가 아니다.다음 단계는 당연히 뿌리 노드의 좌우 나무에 대해 귀속 판단을 해서 잎이 끝날 때까지 하는 것이다.두 갈래 나무 전체의 모든 노드가 판단이 끝났기 때문에 이 문제는 해결되었다.다음은 이러한 사고방식의 구현 코드입니다.

package com.rhwayfun.offer;

public class IsBiTreeBalanced {

    public class TreeNode {
        int val = 0;
        TreeNode left = null;
        TreeNode right = null;

        public TreeNode(int val) {
            this.val = val;

        }

    }

    public boolean IsBalanced_Solution(TreeNode root) {
        if(root == null) return true;
        int leftLen = TreeDepth(root.left);
        int rightLen = TreeDepth(root.right);
        int lenDif = leftLen - rightLen;
        if(lenDif < -1 || lenDif > 1) return false;
        return IsBalanced_Solution(root.left) && IsBalanced_Solution(root.right);
    }

    private int TreeDepth(TreeNode root) {
        if(root == null) return 0;
        int leftLen = TreeDepth(root.left);
        int rightLen = TreeDepth(root.right);
        return leftLen > rightLen ? leftLen + 1 : rightLen + 1;
    }

    public static void main(String[] args) {

    }
}