剑指Offer55-2 平衡二叉树Golang版

剑指Offer55-2 平衡二叉树Golang版

1. 问题描述

输入一棵二叉树的根节点，判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过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 。

限制：

0 <= 树的结点个数 <= 10000

2. 思路

1. 实现一个函数，用于计算二叉树的深度
2. 递归计算每个节点，左子树和右子树高度差，判断绝对值是否大于1

3. 代码

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func isBalanced(root *TreeNode) bool {
    if root == nil {
        return true
    }
    
    if isBalanced(root.Left) == false || isBalanced(root.Right) == false {
        return false
    }

    sub := height(root.Left) - height(root.Right)
    if sub > 1 || sub < -1 {
        return false
    }
    return true
}

func height(p *TreeNode) int {
    if p == nil {
        return 0
    }

    leftHeight := height(p.Left) + 1
    rightHeight := height(p.Right) + 1
    maxHeight := leftHeight
    if rightHeight > maxHeight {
        maxHeight = rightHeight
    }
    return maxHeight
}