LeetCode0110.平衡二叉树 Go语言AC笔记

时间复杂度：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用的左箭头导致编译时报错……所以未来在判断小于一个负数时，一定要记着小于号和负号间空一个空格防止报错！