这是树的第14篇算法,力扣链接。
给定一个二叉树,找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明:叶子节点是指没有子节点的节点。
示例 1:
输入:root = [3,9,20,null,null,15,7] 输出:2
最常规的做法就是层级遍历,当左右子叶都为空跳出当前层级高度就可以了。
func minDepth(root *TreeNode) int {result := 0if root == nil {return result}stack := []*TreeNode{root}result++for len(stack) > 0 {var newStack []*TreeNodefor _, node := range stack {if node.Left == nil && node.Right == nil {return result}if node.Left != nil {newStack = append(newStack, node.Left)}if node.Right != nil {newStack = append(newStack, node.Right)}}result++stack = newStack}return result
}
当然,递归的操作也可以搞,这里值得注意的是,当左右子叶有一个为空并不代表是最小深度,需要将不为空的一边+1返回,这里为了减少判断,直接返回left+right+1了。
func minDepth(root *TreeNode) int {if root == nil {return 0}left, right := minDepth(root.Left), minDepth(root.Right)if root.Left == nil || root.Right == nil {return left + right + 1}if left < right {return left + 1} else {return right + 1}
}