leetcode_110. 平衡二叉树

目录

一、题目内容

二、解题思路

三、代码


一、题目内容

给定一个二叉树,判断它是否是高度平衡的二叉树。

本题中,一棵高度平衡二叉树定义为:

一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过1。

示例 1:

给定二叉树 [3,9,20,null,null,15,7]

    3
   / \
  9  20
    /  \
   15   7
返回 true 。

示例 2:

给定二叉树 [1,2,2,3,3,null,null,4,4]

       1
      / \
     2   2
    / \
   3   3
  / \
 4   4

二、解题思路

记录左右子树的深度,叶节点深度为1,如果累计高度差大于1,则返回小于0的数(这里取-1)。

若两棵子树有一个或两个返回负值,则其父亲也为负。

最后检验root的高度,小于0则False,大于等于0为True。

三、代码


# Definition for a binary tree node.
class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

class Solution:
    def isBalanced(self, root: TreeNode) -> bool:

        def dfs(root):
            if root is None:
                return 0
            lh = dfs(root.left)
            rh = dfs(root.right)
            if lh >= 0 and rh >= 0 and abs(lh - rh) <= 1:
                return 1 + max(lh, rh)
            else:
                return -1
        return dfs(root) >= 0


if __name__ == '__main__':
    a = TreeNode(3)
    a.left = TreeNode(9)
    a.right = TreeNode(20)
    a.right.left = TreeNode(15)
    a.right.right = TreeNode(7)

    s = Solution()
    ans = s.isBalanced(a)
    print(ans)
悲恋花丶无心之人 CSDN认证博客专家 深度学习 神经网络 Pytorch
计算机视觉在读研究生,熟悉Pytorch,MXNet,TensorFlow,Keras等深度学习框架,主要涉及的领域有目标检测,语义分割,超分辨率重建,行人重识别等。
个人GitHub网址为:https://github.com/nickhuang1996
©️2020 CSDN 皮肤主题: Age of Ai 设计师:meimeiellie 返回首页
实付 29.90元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值