leetcode 剑指 Offer 55 - II. 平衡二叉树

题目要求

输入一棵二叉树的根节点，判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过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 get_depth(self, root):
        if not root:
            return 0
        return max(self.get_depth(root.left), self.get_depth(root.right)) + 1

    def isBalanced(self, root):
        """
        :type root: TreeNode
        :rtype: bool
        """
        if not root:
            return True
        depth_left = self.get_depth(root.left)
        depth_right = self.get_depth(root.right)
        if abs(depth_right - depth_left) <= 1 and self.isBalanced(root.left) and self.isBalanced(root.right):
            return True
        return False          

方法二：

1. 对于不是平衡二叉树，返回其高度为-1。
2. 返回用$returnrecursive(root)!=-1$，可以将数值型转换为bool型，从而符合题意。

# 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
        """
        def recursive(root):
            if not root: return 0
            left = recursive(root.left)
            if left == -1: return -1
            right = recursive(root.right)
            if right == -1: return -1
            return max(left, right) + 1 if abs(left-right) <= 1 else -1

        return recursive(root) != -1

知识点

方法二中：返回用$returnrecursive(root)!=-1$，可以将数值型转换为bool型，从而符合题意。