给定一个二叉树,判断它是否是高度平衡的二叉树。
本题中,一棵高度平衡二叉树定义为:
一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 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
}