题目描述:
给定一个二叉树,判断它是否是高度平衡的二叉树。
本题中,一棵高度平衡二叉树定义为:
一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过1。
解题思路:
利用递归求每一棵树的两棵子树的最大深度,若深度绝对值相差>1,则返回false
python代码实现:
# Definition for a binary tree node.
class TreeNode:
def __init__(self, x):
self.val = x
self.left = None
self.right = None
class Solution:
def __init__(self):
self.flag = True
def maxDepth(self, root):
"""
:type root: TreeNode
:rtype: int
"""
if (root == None):#递归终止条件
return 0
else:
leftDepth = self.maxDepth(root.left)
leftDepth = leftDepth + 1
rightDepth = self.maxDepth(root.right)
rightDepth = rightDepth + 1
return max(leftDepth,rightDepth)
def isBalanced(self, root: TreeNode) -> bool:
# 算法思想:先分别求一棵树的两棵子树的最大深度,若深度相差>1,则返回false
if(root == None):
return True
else:
leftDepth = self.maxDepth(root.left)
rightDepth = self.maxDepth(root.right)
num = leftDepth - rightDepth
if(num < 2 and num > -2):
self.flag = self.isBalanced(root.left) and self.isBalanced(root.right)
else:
self.flag = False
return self.flag
n1 = TreeNode(4)
n2 = TreeNode(2)
n3 = TreeNode(7)
n4 = TreeNode(1)
n5 = TreeNode(3)
# n6 = TreeNode(0)
n1.left = n2
n1.right = n3
n2.left = n4
n3.right = n5
# n4.left = n6
s = Solution()
result = s.isBalanced(n1)
print(result)