LeetCode 110 [Balanced Binary Tree]

原题

给定一个二叉树,确定它是高度平衡的。对于这个问题,一棵高度平衡的二叉树的定义是:一棵二叉树中每个节点的两个子树的深度相差不会超过1。

给出二叉树 A={3,9,20,#,#,15,7}, B={3,#,20,15,7}

A)  3            B)    3 
   / \                  \
  9  20                 20
    /  \                / \
   15   7              15  7

二叉树A是高度平衡的二叉树,但是B不是

解题思路

  • 一般BST的问题,首先往divide & conquer的思路去想
  • 一棵二叉树是不是平衡二叉树 => 左子树是不是平衡二叉树?右子树是不是平衡二叉树?
  • 如果任意一颗子树不平衡,整个树肯定不平衡
  • 如果左右子树都是平衡的,但左右子树相差高度大于1,此刻这课树又是不平衡的
  • 问题转化为分别对左右子树求maxDepth,平衡就返回最大深度,不平衡就返回-1
  • 技巧:用-1表示它不是一颗平衡二叉树

完整代码

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

class Solution(object):
    def isBalanced(self, root):
        """
        :type root: TreeNode
        :rtype: bool
        """
        return self.maxDepth(root) != -1
        
    def maxDepth(self, root):
        if not root:
            return 0
        
        left = self.maxDepth(root.left)
        right = self.maxDepth(root.right)
        if left == -1 or right == -1 or abs(left - right) > 1:
            return -1
        else:
            return max(left, right) + 1

你可能感兴趣的:(LeetCode 110 [Balanced Binary Tree])