LintCode 93. Balanced Binary Tree

原创于 2018-02-11 22:41:59 发布 · 191 阅读

0 ·

CC 4.0 BY-SA版权

OJ-LintCode 同时被 2 个专栏收录

85 篇文章

订阅专栏

算法-树

51 篇文章

订阅专栏

本文介绍了一种通过递归方式判断二叉树是否平衡的方法。该算法通过自底向上计算每个节点的高度来确定整棵树是否满足平衡条件，并返回一个布尔值及当前节点的高度。

题目

这里写图片描述

思路

拆解子问题。

代码

"""
Definition of TreeNode:
class TreeNode:
    def __init__(self, val):
        self.val = val
        self.left, self.right = None, None
"""


class Solution:
    """
    @param: root: The root of binary tree.
    @return: True if this Binary tree is Balanced, or false.
    """
    def balance(self, root):
        if not root:
            return True, 0
        leftBool, leftHeight = self.balance(root.left)
        rightBool, rightHeight = self.balance(root.right)
        resBool = False
        if leftBool and rightBool:
            if abs(leftHeight - rightHeight) < 2:
                resBool = True
        return resBool, max(leftHeight, rightHeight) + 1
    def isBalanced(self, root):
        # write your code here
        return self.balance(root)[0]