110.balanced-binary-tree-1.go

/*
 * @lc app=leetcode id=110 lang=golang
 *
 * [110] Balanced Binary Tree
 *
 * https://leetcode.com/problems/balanced-binary-tree/description/
 *
 * algorithms
 * Easy (40.90%)
 * Likes:    1220
 * Dislikes: 108
 * Total Accepted:    319.4K
 * Total Submissions: 779.9K
 * Testcase Example:  '[3,9,20,null,null,15,7]'
 *
 * Given a binary tree, determine if it is height-balanced.
 * 
 * For this problem, a height-balanced binary tree is defined as:
 * 
 * 
 * a binary tree in which the depth of the two subtrees of every node never
 * differ by more than 1.
 * 
 * 
 * Example 1:
 * 
 * Given the following tree [3,9,20,null,null,15,7]:
 * 
 * 
 * ⁠   3
 * ⁠  / \
 * ⁠ 9  20
 * ⁠   /  \
 * ⁠  15   7
 * 
 * Return true.
 * 
 * Example 2:
 * 
 * Given the following tree [1,2,2,3,3,null,null,4,4]:
 * 
 * 
 * ⁠      1
 * ⁠     / \
 * ⁠    2   2
 * ⁠   / \
 * ⁠  3   3
 * ⁠ / \
 * ⁠4   4
 * 
 * 
 * Return false.
 * 
 */
/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
// Note: This solution will compute tree heights redundantly, time complexity is NOT GOOD.
func isBalanced(root *TreeNode) bool {
    if root == nil {
        return true
    }

    leftHeight := treeHeight(root.Left)
    rightHeight := treeHeight(root.Right)

    if abs(leftHeight - rightHeight) > 1 {
        return false
    }
    
    return isBalanced(root.Left) && isBalanced(root.Right)
}

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

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

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