题目链接:https://leetcode-cn.com/problems/balanced-binary-tree/
难度:简单

描述:
给定一个二叉树,判断它是否是高度平衡的二叉树。
本题中,一棵高度平衡二叉树定义为:

  • 一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。

题解

思路:
与104.二叉树的最大深度,只需要比较左右子树高度是否平衡即可。

  1. # Definition for a binary tree node.
  2. # class TreeNode:
  3. # def __init__(self, val=0, left=None, right=None):
  4. # self.val = val
  5. # self.left = left
  6. # self.right = right
  7. class Solution:
  8. def isBalanced(self, root: TreeNode) -> bool:
  9. def recursion(root):
  10. if root is None:
  11. return 0
  12. left_height = recursion(root.left)
  13. if left_height == -1:
  14. return -1
  15. right_height = recursion(root.right)
  16. if right_height == -1:
  17. return -1
  18. # 若平衡返回以该节点为根的树的高度
  19. # 若不平衡返回-1
  20. if abs(left_height - right_height) < 2:
  21. return max(left_height, right_height) + 1
  22. else:
  23. return -1
  24. return recursion(root) != -1