目录
问题描述:
给定一个二叉树,找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明:叶子节点是指没有子节点的节点。
示例 1:
输入:root = [3,9,20,null,null,15,7] 输出:2
示例 2:
输入:root = [2,null,3,null,4,null,5,null,6] 输出:5
实现代码与解析:
递归:
容易写成的错误代码:
class Solution {
public:
int minDepth(TreeNode* root)
{
if(root==NULL) return 0;
return min(minDepth(root->left),minDepth(root->right))+1;
}
};
错误分析:
很多人会把求最大深度的题改一下用在这里,其实是不对的,最小深度是到叶子结点的最小距离,如果只是把求最大深度的max改成min,会在一些情况下把左子树或右子树为空结点的0深度加1返回(如下图),显然是不对的,所以我们要进行代码的修改,在if返回条件的地方。
正确代码:
class Solution {
public:
int minDepth(TreeNode* root)
{
if(root==NULL)
{
return 0;
}
int left=minDepth(root->left);
int right=minDepth(root->right);
if(root->left==NULL&&root->right!=NULL) return 1+right;//左子树为空,右子树不空
else if(root->left!=NULL&&root->right==NULL) return 1+left;//右子树为空,左子树不空 //都不空
return 1+min(left,right);
}
};
原理思路:
我们在错误代码的基础上判断返回的时候判断一下,若左右子树其中一个为空,我们就返回对应相反子树的深度加一。
层序遍历:
class Solution {
public:
int minDepth(TreeNode* root)
{
queue<TreeNode*> que;
if(root==NULL) return 0;
que.push(root);
int result=0;
while(!que.empty())
{
int size=que.size();
result++;
for(int i=0;i<size;i++)
{
TreeNode* temp=que.front();
que.pop();
if(temp->left) que.push(temp->left);
if(temp->right) que.push(temp->right);
//当左右子树为空时返回
if(temp->left==NULL&&temp->right==NULL)
{
return result;
}
}
}
return result;
}
};
原理思路:
基于层序遍历的代码基础上,添加一个记录层数的,每进入一层就加一,然后判断一下结点左右子树是否为空,若都为空就说明到了第一个有叶子结点的层,也就是最小深度。
还有一个要注意的点,最后的return一定要写,不然会编译错误,虽然我觉得一定会在if里返回,不会走到这里。
查了查,应该是虽然我们if里考虑了所有情况一定能返回,但是编译器不这么认为,所以要加上最后的return,其实我还是不太能理解,有没有大佬解答一下,或许必须要有一个return要脱离if?。