刷题--程序员面试金典--面试题 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
}