1、二叉树判断平衡树问题,参考于leetcode110,给定一颗二叉树,判断它是否是高度平衡的二叉树,定义为,一个二叉树每个节点的左右两个子树的高度差的绝对值不超过1,也就是循环遍历每个节点,如果左右子树不平衡直接返回-1,平衡则继续判断。
class Solution:
def isBalanced(self, root: Optional[TreeNode]) -> bool:
def height(root):
if not root:
return 0
leftheight = height(root.left)
rightheight = height(root.right)
if abs(leftheight - rightheight) > 1 or leftheight == -1 or rightheight == -1:
return -1
else:
return 1 + max(leftheight, rightheight)
return height(root) >= 0
2、二叉树最小深度问题,参考于leetcode111,给定一颗二叉树,找出其最小深度,最小深度是从根节点到最近叶子结点的最短路径上的数量,不能直接把最大深度搬过来,核心问题是分析终止条件:如果左子树为空,右子树不为空,说明最小深度是1 + 右子树的深度,反之,如果右子树为空,左子树不为空,说明最小深度是1 + 左子树的深度,如果左右子树都不为空,返回左右深度最小值+1
class Solution:
def minDepth(self, root: Optional[TreeNode]) -> int:
if not root:
return 0
if not root.left and not root.right:
return 1
if not root.right:
return self.minDepth(root.left) + 1
if not root.left:
return self.minDepth(root.right) +1
return min(self.minDepth(root.left), self.minDepth(root.right)) + 1