LeetCode110. 平衡二叉树Golang版

最新推荐文章于 2024-06-09 21:18:28 发布

原创最新推荐文章于 2024-06-09 21:18:28 发布 · 187 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树 #算法 #leetcode #面试

leetcode刷题专栏收录该内容

252 篇文章

订阅专栏

本文介绍LeetCode110题目的解决方案，通过递归方式判断一棵二叉树是否为高度平衡二叉树，并提供Golang实现代码。

LeetCode110. 平衡二叉树Golang版

1. 问题描述

给定一个二叉树，判断它是否是高度平衡的二叉树。

本题中，一棵高度平衡二叉树定义为：

一个二叉树每个节点的左右两个子树的高度差的绝对值不超过 1 。
在这里插入图片描述

2. 思路

2.1. 递归

确定递归函数参数和返回值

	func isBalanced(root *TreeNode) bool

确定结束条件

 	if root == nil {
        return true
    }

确定单层递归逻辑

 	l := height(root.Left)
    r := height(root.Right)

    if l + 1 < r || r + 1 < l {
        return false
    }
    res := isBalanced(root.Left)&&isBalanced(root.Right)
    return res

3. 代码

3.1. 递归代码

/**
 * 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
    }
    
    l := height(root.Left)
    r := height(root.Right)

    if l + 1 < r || r + 1 < l {
        return false
    }

    return isBalanced(root.Left)&&isBalanced(root.Right)
}

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

    l := height(root.Left)
    r := height(root.Right)

    if l < r {
        l = r
    }
    return l + 1
}