这是树的第14篇算法,力扣链接。
给定一个二叉树,找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明:叶子节点是指没有子节点的节点。
示例 1:
输入:root = [3,9,20,null,null,15,7] 输出:2
最常规的做法就是层级遍历,当左右子叶都为空跳出当前层级高度就可以了。
func minDepth(root *TreeNode) int {
result := 0
if root == nil {
return result
}
stack := []*TreeNode{root}
result++
for len(stack) > 0 {
var newStack []*TreeNode
for _, 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
}
}