求解二叉树的最小深度,即可以递归实现,也可以利用BFS非递归实现
1.递归版本:
注意条件的判断:
叶子节点的定义是左孩子和右孩子都为 null 时叫做叶子节点
当 root 节点左右孩子都为空时,返回 1
当 root 节点左右孩子有一个为空时,返回不为空的孩子节点的深度
当 root 节点左右孩子都不为空时,返回左右孩子较小深度的节点值
public int minDepth(TreeNode root) {
if (root == null) return 0;
int left = minDepth(root.left);
int right = minDepth(root.right);
return root.left == null || root.right == null ? left + right + 1 : Math.min(left, right) + 1;
}
2.BFS版本:
public int minDepth(TreeNode root) {
if (root == null) return 0;
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
int level = 1;
while (!queue.isEmpty()) {
int size = queue.size();
while (size-- > 0) {
TreeNode top = queue.poll();
if (top.left == null && top.right == null) return level;
if (top.left != null) queue.offer(top.left);
if (top.right != null) queue.offer(top.right);
}
level++;
}
return level;
}
网友评论