题目描述
给定一个二叉树,找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
**说明:**叶子节点是指没有子节点的节点。
示例 1:
输入:root = [3,9,20,null,null,15,7]
输出:2
示例 2:
输入:root = [2,null,3,null,4,null,5,null,6]
输出:5
提示:
- 树中节点数的范围在
[0, 105]
内 -1000 <= Node.val <= 1000
解答
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
int minDepth(TreeNode* root) {
// 后序遍历二叉树求深度
// 以某一节点为根的二叉树最小深度是:左右子树深度小者 + 1
return getDepth(root);
}
private:
int getDepth(TreeNode* root)
{
if(root == nullptr) return 0;
int leftDepth = getDepth(root->left);
int rightDepth = getDepth(root->right);
// 左子树空,右子树不空,最小深度必然在右子树(叶节点出现在右子树)
if(root->left == nullptr && root->right != nullptr)
{
return 1 + rightDepth;
}
if(root->left != nullptr && root->right == nullptr)
{
return 1 + leftDepth;
}
// 左右子树都不空,选小者
return 1 + min(leftDepth, rightDepth);
}
};