平衡二叉树
概述:给定一个二叉树,判断它是否是高度平衡的二叉树。一棵高度平衡二叉树定义为:一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。
输入:root = [3,9,20,null,null,15,7]
输出:true
输入:root = [1,2,2,3,3,null,null,4,4]
输出:false
输入:root = []
输出:true
方法一:自顶向下的递归
思路:定义函数 height,用于计算二叉树中的任意一个节点 p 的高度。对于当前遍历到的节点,首先计算左右子树的高度,如果左右子树的高度差是否不超过 1 ,再分别递归地遍历左右子节点,并判断左子树和右子树是否平衡。
# 自顶向下的递归
class Solution:
def isBalanced(self, root: Optional[TreeNode]) -> bool:
def height(root: TreeNode) -> int:
if not root:
return 0
return max(height(root.left), height(root.right)) + 1
if not root:
return True
return abs(height(root.left) - height(root.right)) <= 1 \
and self.isBalanced(root.left) and self.isBalanced(root.right)
方法二:自底向上的递归
思路:自底向上递归的做法类似于后序遍历,对于当前遍历到的节点,先递归地判断其左右子树是否平衡,再判断以当前节点为根的子树是否平衡。
# 自底向上的递归
class Solution:
def isBalanced(self, root: Optional[TreeNode]) -> bool:
def height(root: TreeNode) -> int:
if not root:
return 0
left_height = height(root.left)
right_height = height(root.right)
if left_height == -1 or right_height == -1 \
or abs(left_height - right_height) > 1:
return -1
else:
return max(left_height, right_height) + 1
return height(root) >= 0
总结
自底向上的递归就是及时止损,自顶向下的递归就是不管三七二十一先把所有的高度算一遍!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。
如需转载请保留出处:https://bianchenghao.cn/35915.html