当前位置: 代码迷 >> 综合 >> LeetCode: 110.平衡二叉树
  详细解决方案

LeetCode: 110.平衡二叉树

热度:67   发布时间:2023-10-20 18:42:04.0

上一道题是建立一个高度平衡二叉树,下一道题目就是判断一棵树是否是平衡二叉树。

平衡二叉树的定义是:

一个树的每一个节点作为根节点,其左右子树的高度相差小于等于1。树的高度是指树的最大深度。

 

惭愧,我也是看了题解才明白的。然后自己实现了暴露法,和题解写法不同。我们一一来看。

有两种方法。第一种方法就是自顶而下深度优先遍历,先求出节点的左子树和右子树的高度,然后判断高度差是否满足条件,如果满足,则说明以这个节点为根节点的树是高度平衡的,进而判断这个节点的左节点和右节点是否平衡的,如果不满足,返回False。然后使用递归遍历每一个节点。

class Solution:def isBalanced(self, root: TreeNode) -> bool:if root == None:return Truedef depth(root, num):if root == None:return num-1else:return max(depth(root.left, num+1), depth(root.right, num+1))if abs(depth(root.right, 2) - depth(root.left, 2)) <= 1:return self.isBalanced(root.left) and self.isBalanced(root.right)else:return False

我的官方题解写法是不同的。我注意到官方写法,求树深度是这样写的

def depth(root):if root == None:return 0else:return max(depth(root.right), depth(root.left)) + 1

观察得知,我通过把高度写到函数参数列表中和题解使用返回值+1的形式,都能得到目的,学到了。

 

第二种更优的办法是,从底至顶深度优先遍历,也称为提前阻断法,如果底层的节点都不满足是平衡二叉树了,那整棵树都不会是了,就可以提前退出了,顾称提前阻断法。

class Solution:def isBalanced(self, root: TreeNode) -> bool:if root == None:return Trueif self.depth(root) == -1:return Falseelse:return Truedef depth(self, root):if not root:return 0left = self.depth(root.left)if left == -1:return -1right = self.depth(root.right)if right == -1:return -1if abs(left - right) >1:return -1else:return max(left, right) +1 

区别于第一种方法,就是在递归的时候,一直遍历到底层,加了一个判断条件,判断底层节点是否是满足平衡二叉树的定义。不满足则返回 -1,递归返回-1。