刷題--程序員面試金典--面試題 04.04. 檢查平衡性（go）

面試題 04.04. 檢查平衡性

實現一個函數，檢查二叉樹是否平衡。在這個問題中，平衡樹的定義如下：任意一個節點，其兩棵子樹的高度差不超過 1。

示例 1:
給定二叉樹 [3,9,20,null,null,15,7]
    3
   / \
  9  20
    /  \
   15   7
返回 true 。
示例 2:
給定二叉樹 [1,2,2,3,3,null,null,4,4]
      1
     / \
    2   2
   / \
  3   3
 / \
4   4
返回 false 。

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/check-balance-lcci
著作權歸領釦網絡所有。商業轉載請聯繫官方授權，非商業轉載請註明出處。

---

思路：

核心是得到二叉樹的最大深度，並判斷左右子樹的深度差。

func isBalanced(root *TreeNode) bool {
    if root == nil {
        return true
    }
    flag := true
    run(root,&flag)

    return flag
}

func run(root *TreeNode,flag *bool) int {
    if *flag == false {
        return 0
    }

    if root == nil {
        return 1
    }


    l := run(root.Left,flag)
    r := run(root.Right,flag)

    if abs(l-r) > 1 {
        *flag = false
    }

    if l > r {
        return l+1
    }
    return r+1
}

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