二叉树的最短路径

最新推荐文章于 2023-05-10 10:10:00 发布

不爱码字的程序鸽

最新推荐文章于 2023-05-10 10:10:00 发布

阅读量1.1k

点赞数

文章标签：二叉树

本文链接：https://blog.csdn.net/qq_41818406/article/details/104546643

版权

1.涉及到最短路径要用二叉树的广度优先搜索
用到队列

import javafx.util.Pair;
class Solution {
  public int minDepth(TreeNode root) {
    LinkedList<Pair<TreeNode, Integer>> stack = new LinkedList<>();
    if (root == null) {
      return 0;
    }
    else {
      stack.add(new Pair(root, 1));
    }

    int current_depth = 0;
    while (!stack.isEmpty()) {
      Pair<TreeNode, Integer> current = stack.poll();
      root = current.getKey();
      current_depth = current.getValue();
      if ((root.left == null) && (root.right == null)) {
        break;
      }
      if (root.left != null) {
        stack.add(new Pair(root.left, current_depth + 1));
      }
      if (root.right != null) {
        stack.add(new Pair(root.right, current_depth + 1));
      }
    }
    return current_depth;
  }
}

步骤：
1.用到Pair<,> 这个数据结构。
2.根节点进栈
3.while 不为空
4.出站
5.左子树右子树进栈