时间复杂度:O(n)
解题思路
借助于后序遍历,每次递归遍历返回一个高度值,如果高度值不小于0说明是正常高度,-1代表是失去平衡。这样通过递归遍历左子树和右子树分别得到两个高度值,如果左子树高度或右子树高度为-1或者左右子树高度差大于1,就返回-1,说明左右子树高度已经失衡;否则返回两子树的最大高度加1,表示该结点的高度。
AC代码
/**
* Definition for a binary tree node.
* type TreeNode struct {
* Val int
* Left *TreeNode
* Right *TreeNode
* }
*/
func isBalanced(root *TreeNode) bool {
var dfs func(*TreeNode)int
dfs=func(root *TreeNode)int{
if root==nil{
return 0
}
left:=dfs(root.Left)
right:=dfs(root.Right)
//判断小于-1时,小于号和负号间一定要有一个空格,否则会认为是channel赋值
if left==-1||right==-1||left-right>1||left-right< -1 {
return -1
}
return max(left,right)+1
}
return dfs(root)>-1
}
func max(x,y int)int{
if x>y{
return x
}
return y
}
感悟
只想到了返回高度值,但是没有想到可以借助于-1表示左右子树失衡。最后返回-1的判断语句又报错,找来找去才发现原来是因为小于号和负号连用形成了channel用的左箭头导致编译时报错……所以未来在判断小于一个负数时,一定要记着小于号和负号间空一个空格防止报错!