Leetcode 111. 二叉树的最小深度
广度优先
class Solution {
// 内部类,记录节点信息和当前节点的深度
static class QueueNode {
TreeNode treeNode;
int depth;
public QueueNode(TreeNode node, int depth) {
this.treeNode = node;
this.depth = depth;
}
}
public int minDepth(TreeNode root) {
if (root == null) {
return 0;
}
// 使用队列记录节点信息
Queue<QueueNode> queue = new LinkedList<>();
queue.offer(new QueueNode(root, 1));
while (!queue.isEmpty()) {
QueueNode queueNode = queue.poll();
TreeNode treeNode = queueNode.treeNode;
int depth = queueNode.depth;
// 如果treeNode的左节点与右节点都未空,则当前节点的深度就是整棵树的最小深度
if (treeNode.left == null && treeNode.right == null) {
return depth;
}
if (treeNode.left != null) {
queue.offer(new QueueNode(treeNode.left, depth + 1));
}
if (treeNode.right != null) {
queue.offer(new QueueNode(treeNode.right, depth + 1));
}
}
return 0;
}
}
深度优先
class Solution {
public int minDepth(TreeNode root) {
return dfs(root);
}
private int dfs(TreeNode root) {
if (root == null) {
return 0;
}
// 没有子节点,返回当前层
if (root.left == null && root.right == null) {
return 1;
}
int minDepth = Integer.MAX_VALUE;
if (root.left != null) {
// 递归计算左子树的最小深度
minDepth = Math.min(dfs(root.left), minDepth);
}
if (root.right != null) {
// 递归计算右子树的最小深度
minDepth = Math.min(dfs(root.right), minDepth);
}
// 返回最小深度(返回时需要加上当前层)
return minDepth + 1;
}
}