剑指 Offer 55 - II. 平衡二叉树

和力扣 110. 平衡二叉树一致
给定一个二叉树,判断它是否是高度平衡的二叉树。
本题中,一棵高度平衡二叉树定义为:

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


示例 1:
剑指 55 - II. 平衡二叉树 ✂️ - 图1
输入:root = [3,9,20,null,null,15,7]
输出:true

  1. //正解,自下而上的递归,时空On
  2. func isBalanced(root *TreeNode) bool {
  3. return height(root) >= 0
  4. }
  5. func height(root *TreeNode) int {
  6. if root == nil {
  7. return 0
  8. }
  9. LeftHight := height(root.Left)
  10. RightHight := height(root.Right)
  11. if abs(LeftHight - RightHight) > 1 || LeftHight == -1 || RightHight == -1 {
  12. return -1
  13. }
  14. return max(LeftHight, RightHight) + 1
  15. }
  16. func max(x, y int) int {
  17. if x > y {
  18. return x
  19. }
  20. return y
  21. }
  22. func abs(x int) int {
  23. if x < 0 {
  24. return -1 * x
  25. }
  26. return x
  27. }
//自上而下递归, 最坏时间On^2,平均时间Onlogn, 空间on
func isBalanced(root *TreeNode) bool {
    if root == nil {
        return true 
    }
    return abs(height(root.Left) - height(root.Right)) <= 1 && isBalanced(root.Left) && isBalanced(root.Right)
}

func height(root *TreeNode) int {
    if root == nil {
        return 0
    }
    return max(height(root.Left) , height(root.Right)) + 1
}

func abs(x int) int {
    if x < 0 {
        return -1 * x
    }
    return x
}

func max(x, y int) int {
    if x > y {
        return x
    }
    return y
}