Leetcode111. 二叉树的最小深度

Leetcode111. 二叉树的最小深度

题目：
给定一个二叉树，找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明: 叶子节点是指没有子节点的节点。

示例:
给定二叉树 [3,9,20,null,null,15,7]，
     3
    / \
   9  20
     /  \
    15   7
返回它的最小深度  2.

题解：
scala代码如下：
递归：

/**
    * 递归
    *
    * @param root
    * @return
    */
  def minDepth(root: TreeNode): Int = {
    if (root == null) {
      0
    } else if (root.left == null && root.right == null) {
      1
    } else {
      var min_depth = Int.MaxValue
      if (root.left != null) {
        min_depth = Math.min(min_depth, minDepth(root.left))
      }
      if (root.right != null) {
        min_depth = Math.min(min_depth, minDepth(root.right))
      }
      min_depth + 1
    }
  }

迭代：

/**
    * 迭代
    *
    * @param root
    * @return
    */
  def minDepth2(root: TreeNode): Int = {
    var flag = true
    val stack: util.Stack[(TreeNode, Int)] = new util.Stack[(TreeNode, Int)]()
    var min_depth = Int.MaxValue
    if (root == null) {
      flag = false
      min_depth = 0
    } else {
      stack.add((root, 1))
    }
    if (flag) {
      while (!stack.empty()) {
        val tuple = stack.pop()
        val root = tuple._1
        val current_depth = tuple._2
        if (root.left == null && root.right == null) {
          min_depth = Math.min(min_depth, current_depth)
        }
        if (root.left != null) {
          stack.add((root.left, current_depth + 1))
        }
        if (root.right != null) {
          stack.add((root.right, current_depth + 1))
        }
      }
    }
    min_depth
  }

java代码：

   public static int minDepth(TreeNode root) {
        if (root == null) return 0;
        Stack<Pair<TreeNode, Integer>> stack = new Stack<>();

        stack.push(new Pair<>(root, 1));
        int depth = Integer.MAX_VALUE;
        while (!stack.isEmpty()) {
            Pair<TreeNode, Integer> pop = stack.pop();
            TreeNode node = pop.getKey();
            Integer value = pop.getValue();
            if (node.left == null && node.right == null) {
                depth = Math.min(depth, value);
            }

            if (node.left != null) stack.push(new Pair<>(node.left, value + 1));
            if (node.right != null) stack.push(new Pair<>(node.right, value + 1));
        }
        return depth;
    }