LeetCode 111 二叉树的最小深度

还是递归, 简单难度, 本质上是一种遍历, 唯一要注意的就是 只有 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 可以用来表示 "无穷" 的概念.