coding/알고리즘,자료구조

leetcode - 균형 이진트리(balanced binary tree)

Jeo 2022. 1. 27. 14:58

이진트리야 친해지자..^^

재귀를 쓰더라도 부모 자신부터 처리하고 자식레벨로 넘어가는 상황은 조금이나마 직관적으로 받아들이기가 낫다.

그런데 이 문제처럼 아예 리프 노드부터 처리되고 그 리턴값을 받아와서 그걸로 그 상단 노드가 처리를 해야 할 경우가 아직 어렵다..^^

그래도 요런 아이들도 자꾸 보다보면 익숙해지겠지라는 마음으로 공부공부✍️

1. 손정리

class Solution(object):
    def isBalanced(self, root):
        def DFS(root):
            if not root:
                return 0
            
            left = DFS(root.left)
            right = DFS(root.right)
            
            # 균형이 깨졌음을 알려줄 수 있는 세 가지 지표
            if abs(left-right) > 1 or left == -1 or right == -1:
                return -1
            
            return max(left, right) + 1
        
        return DFS(root) != -1