LeetCode:111.二叉树的最小深度
给定一个二叉树,找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明:叶子节点是指没有子节点的节点。
示例 1:
输入:root = [3,9,20,null,null,15,7]
输出:2
示例 2:
输入:root = [2,null,3,null,4,null,5,null,6]
输出:5
递归,类似后序遍历,整体的处理逻辑和上一题基本一致,只是需要注意子节点为空的情况是如何处理的
public int minDepth(TreeNode root) {
if (root == null)
return 0;
int leftHeight = minDepth(root.left);
int rightHeight = minDepth(root.right);
if (root.left == null && root.right != null) {
return 1 + rightHeight;
} else if (root.left != null && root.right == null) {
return 1 + leftHeight;
} else {
return 1 + Math.min(leftHeight, rightHeight);
}
}
精简版
public int minDepth(TreeNode root) {
if (root == null)
return 0;
if (root.left == null && root.right != null) {
return 1 + minDepth(root.right);
} else if (root.left != null && root.right == null) {
return 1 + minDepth(root.left);
} else {
return 1 + Math.min(minDepth(root.left), minDepth(root.right));
}
}