法一:自底向上递归
解题思路:
每层调用递归栈结束(返回节点高度)的条件为当前节点高度为-1或者|左子树高度-右子树高度|大于1;
首先从根结点开始,会通过height函数进入到最深的左子树,然后从下往上传递节点的高度,在这个传递的过程也会计算右节点的高度,返回的高度一定是不大于1的正数,如果存在一棵子树不平衡,则整个二叉树一定不平衡,即会返回-1。
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(x int)int{
if x<0{
return -1*x
}
return x
}
时间复杂度:O(n) ,n为二叉树节点个数,每个节点都会遍历一遍
空间复杂度:O(n),n为二叉树节点个数,每个节点都会调用一次递归栈。