LeetCode 110 平衡二叉树

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

本题中,一棵高度平衡二叉树定义为:

一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。

示例 1:

输入:root = [3,9,20,null,null,15,7]
输出:true

示例 2:

输入:root = [1,2,2,3,3,null,null,4,4]
输出:false

示例 3:

输入:root = []
输出:true

提示:

  • 树中的节点数在范围 [0, 5000] 内
  • -10^4 <= Node.val <= 10^4

解题:

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

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

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

func isBalanced(root *TreeNode) bool {
	if root == nil {
		return true
	}
	return abs(getHeight(root.Left)-getHeight(root.Right)) <= 1 && isBalanced(root.Left) && isBalanced(root.Right)
}

官方解答:

1.自顶向下的递归

func isBalanced(root *TreeNode) bool {
	if root == nil {
		return true
	}
	return abs(height(root.Left)-height(root.Right)) <= 1 && isBalanced(root.Left) && isBalanced(root.Right)
}

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

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

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

2.自底向上的递归

func isBalanced(root *TreeNode) bool {
	return height(root) >= 0
}

func height(root *TreeNode) int {
	if root == nil {
		return 0
	}
	leftHeight := height(root.Left)
	rightHeight := height(root.Right)
	if leftHeight == -1 || rightHeight == -1 || abs(leftHeight-rightHeight) > 1 {
		return -1
	}
	return max(leftHeight, rightHeight) + 1
}

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

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

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注