题解 | #判断是不是平衡二叉树# | Golang

package main
import . "nc_tools"
/*
 * type TreeNode struct {
 *   Val int
 *   Left *TreeNode
 *   Right *TreeNode
 * }
 */

/**
 * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
 *
 * 
 * @param pRoot TreeNode类 
 * @return bool布尔型
*/
func IsBalanced_Solution( pRoot *TreeNode ) bool {
    if pRoot == nil {
        return true
    }
    return IsBalanced_Solution(pRoot.Left) && IsBalanced_Solution(pRoot.Right) && abs(Height(pRoot.Left) - Height(pRoot.Right)) <= 1
}

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

func max(a int, b int) int {
    if a > b {
        return a
    }
    return b
}

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