给定一个二叉树,找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明:叶子节点是指没有子节点的节点。
//本题有一个大大的坑,就是当根节点只有左孩子,没有右孩子的情况,我们就返回左子树的最小深度(或者只有右孩子,没有左孩子,我们就返回右子树的最小深度)
//因为最小深度是根节点到叶子节点,深度最小的情况,而叶子节点是指没有左右孩子的节点,如果测试用来如如上图中所示情况,它的最小深度为3,为什么不是呢,因为根节点有右孩子,不属于叶子节点
class Solution {
public:
int getDepth(TreeNode* root) {
//递归的结束条件
if(root == nullptr){
return 0;
}
//判断左子树的深度
int leftDepth = getDepth(root->left);
//判断左子树的深度
int rightDepth = getDepth(root->right);
//考虑此时递归的节点,如果左孩子为空,右孩子不为空的情况,说明此递归的节点非叶子节点,返回右子树的深度+1(1表示根节点)
if(root->left == nullptr&&root->right != nullptr) {
return rightDepth + 1;
}
//考虑此时递归的节点,如果左孩子不为空,右孩子为空的情况,说明此递归的节点非叶子节点,返回左子树的深度+1(1表示根节点)
else if(root->left != nullptr&&root->right == nullptr) {
return leftDepth + 1;
}
//如果此时递归的节点,、左孩子与右孩子均不为空的情况,返回较小者+1(1表示根节点)
return min(leftDepth,rightDepth) + 1;
}
int minDepth(TreeNode* root) {
//如果传进来的根节点为空的话,直接返回0即可
if(root == nullptr){
return 0;
}
//如果传进来的根节点左右孩子均为空的话,直接返回1即可,此时根节点可看作叶子节点,所以最小深度为1
else if(root->left == nullptr && root->right == nullptr) {
return 1;
}
//其余情况,依函数判断
return getDepth(root);
}
};