还是递归, 简单难度, 本质上是一种遍历, 唯一要注意的就是 只有 node 的左右结点均为 null 的时候才需要去更新最小值
def minDepth(root: TreeNode): Int = {
var min : Option[Int] = None
def go(node: TreeNode, level: Int = 0): Unit = {
if (node != null) {
if(node.left == null && node.right == null){
min = Some(min.map(math.min(level,_)).getOrElse(level))
} else if(min.isEmpty || level < min.get - 1){
go(node.left, level + 1)
go(node.right, level + 1)
}
}
}
if (root == null) 0
else {
go(root)
min.get
}
}
这里有个小技巧, min变量使用 None 而不使用 Int.MinValue , 是因为 None 或者 null 可以用来表示 "无穷" 的概念.