1. 题目描述
题目来源:力扣
给定一个二叉树,判断它是否是高度平衡的二叉树。
本题中,一棵高度平衡二叉树定义为:
一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1
2. 题解
2.1 题解思路
从根节点开始,判断一个节点的左右深度差是否超过1。分析:从根节点开始,分别获取左右子树的高度,那么根节点的左子节点也需判断其左右子树的高度差是否超过1,直到叶子节点位置。且根节点的左子树的高度为 max{左子树的左子树高度,左子树的右子树高度}。根据该分析,可知判断一棵树是否为平衡树的逻辑思路为:递归结构。
按照上述思路,返回值为一个节点左右子树的高度。我们想要得到一棵树是否为平衡树,那么上述过程可以作为一个子函数调用。难点在于,每个节点处判断左右子树高度差是否超过1。如果超过1,标记返回结果-1。如果当前节点的左子树的高度为-1,返回上一级函数时,仍返回-1;右子树同样。
按照此思路,算法如下:
内层函数:从根节点开始
(1)初始化当前节点的左右子度为Level_left = Level_right = 0
(2)递归调用左子节点的高度,若左子树的高度为-1,返回-1;
(3)递归调用右子节点的高度,若右子树的高度为-1,返回-1;
(4)若|左子树高度-右子树的高度| > 1,那么返回-1;
(5)返回该节点的深度为max{左子树的高度,右子树的高度}+1
外层函数:最后判断返回值如果为-1,那么不是平衡树,否则是平衡树。
2.2 代码
代码如下,
# Definition for a binary tree node.
class TreeNode(object):
def __init__(self, val=0, left=None, right=None):
self.val = val
self.left = left
self.right = right
def set_left(self, left):
self.left = left
def set_right(self, right):
self.right = right
class Solution(object):
def node_level(self, root):
"""
:param node: TreeNode
:return: node的左右深度
"""
current = root
L_left = 0
L_right = 0
if current.left != None:
L_left = self.node_level(current.left)
if L_left == -1:
return -1
if current.right != None:
L_right = self.node_level(current.right)
if L_right == -1:
return -1
if L_right - L_left > 1 or L_right - L_left < -1:
return -1
else:
return max(L_left, L_right)+1
def isBalanced(self, root):
"""
:type root: TreeNode
:rtype: bool
"""
if root == None:
return True
if self.node_level(root) == -1:
return False
else:
return True
if __name__ == '__main__':
root = [1, 2, 2, 3, 4, 4, 3]
root = [1, 2, 3]
node1 = TreeNode(1)
node2 = TreeNode(2)
node3 = TreeNode(3)
node4 = TreeNode(4)
node5 = TreeNode(5)
node6 = TreeNode(6)
node7 = TreeNode(7)
node1.set_left(node2)
node1.set_right(node3)
node2.set_left(node4)
node4.set_left(node6)
node3.set_right(node5)
node5.set_right(node7)
sol = Solution()
r_level = sol.node_level(node1)
r = sol.isBalanced(node1)
print(r_level)
print(r)
这其中有一个细节,l_left和l_right初始化为0还是1,然后在左节点非空的情况下,L_left是直接+1还是在判断完L_left==-1;应该是初始化为0,然后在判断子树及本身不是非平衡树的情况下,再—+1.