110.平衡二叉树

Author: techoc

README.md

@@ -25,6 +25,7 @@
 - [101.对称二叉树](leetcode/101.go)
 - [104.二叉树的最大深度](leetcode/104.go)
 - [108.将有序数组转换为二叉搜索树](leetcode/108.go)
+- [110.平衡二叉树](leetcode/110.go)
 - [111.二叉树的最小深度](leetcode/111.go)
 - [112.路径总和](leetcode/112.go)
 - [118.杨辉三角](leetcode/118.go)

leetcode/110.go

@@ -0,0 +1,29 @@
+package leetcode
+
+import "math"
+
+//给定一个二叉树，判断它是否是高度平衡的二叉树。
+//本题中，一棵高度平衡二叉树定义为：
+//一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。
+
+func isBalanced(root *TreeNode) bool {
+	if root == nil {
+		return true
+	}
+	var getDepth func(root *TreeNode) int
+	getDepth = func(root *TreeNode) int {
+		if root == nil {
+			return 0
+		}
+		left := getDepth(root.Left)
+		right := getDepth(root.Right)
+		//返回节点左右子树的最大深度+1
+		return max(left, right) + 1
+	}
+
+	if math.Abs(float64(getDepth(root.Left)-getDepth(root.Right))) > 1 {
+		return false
+	}
+	// 左右子树都是平衡二叉树
+	return isBalanced(root.Left) && isBalanced(root.Right)
+}