力扣111二叉树的最小深度(c++)
思路
本题采用前序遍历和后序遍历都可以,前序求的是深度,后序求的是高度。
二叉树节点的深度:指从根节点到该节点的最长简单路径边的条数或者节点数(取决于深度从0开始还是从1开始)
二叉树节点的高度:指从该节点到叶子节点的最长简单路径边的条数后者节点数(取决于高度从0开始还是从1开始)
那么使用后序遍历,其实求的是根节点到叶子节点的最小距离,就是求高度的过程,而且这个最小距离,也同样是最小深度。
本文给出的代码是采用后序遍历的递归方法
本题还有一个误区,注意题目中说:最小深度是从根节点到最近叶子节点的最短路径上的节点数量,注意是叶子节点。
什么是叶子节点,左右孩子都为空的节点才是叶子节点!
来看看递归三部曲:
1.确定递归函数的参数和返回值
参数为要传入的二叉树根节点,返回的是int类型的深度。
2.确定终止条件
终止条件也是遇到空节点返回0,表示当前节点的高度为0。
3.确定单层递归的逻辑
如果左子树为空,右子树不为空,说明最小深度是 1 + 右子树的深度。
反之,右子树为空,左子树不为空,最小深度是 1 + 左子树的深度。
最后如果左右子树都不为空,返回左右子树深度最小值 + 1 。
代码如下
class Solution
{
public:
int getDepth(TreeNode* node){ //后序遍历
if (node==NULL) //终止条件,遇到空节点返回0,表示当前节点的高度为0
return 0;
int leftDepth = getDepth(node->left); //左
int rightDepth = getDepth(node->right); //右
//中
//当一个左子树为空,右不为空,这时并不是最低点
if (node->left==NULL&&node->right!=NULL)
return 1 + rightDepth;
//当一个右子树为空,左不为空,这时并不是最低点
if (node->left!=NULL&&node->right==NULL)
return 1 + leftDepth;
int result = 1 + min(leftDepth, rightDepth);
return result;
}
int minDepth(TreeNode *root)
{
return getDepth(root);
}
};
代码运行结果