题目链接: 二叉树的最小深度
有关题目
给定一个二叉树,找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明:叶子节点是指没有子节点的节点。
示例 1:
输入:root = [3,9,20,null,null,15,7]
输出:2
示例 2:
输入:root = [2,null,3,null,4,null,5,null,6]
输出:5
提示:
树中节点数的范围在 [0, 10^5] 内
-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) {
if (root == nullptr)//递归结束条件
return 0;
if (root->left == nullptr && root->right == nullptr)//叶子节点
return 1;
int min_depth = INT_MAX;//求孩子节点的最小深度
if (root->left != nullptr)
min_depth = min(min_depth, minDepth(root->left));
if (root->right != nullptr)
min_depth = min(min_depth, minDepth(root->right));
return 1 + min_depth;//当前节点的最小深度为孩子节点的最小深度 加 一
}
};
法二:广度优先搜索
参考官方题解
/**
* 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) {
if (root == nullptr)
return 0;
//利用队列的先进先出的特性,最先搜索到的叶子节点的深度一定最小
//优先队列中的元素存放的为当前节点 与 当到这个节点的的深度
queue<pair<TreeNode *, int>> q;
q.emplace(root, 1);
while(!q.empty())
{
TreeNode* node = q.front().first;//获得当前节点
int depth = q.front().second;//获得遍历到当前节点的深度
q.pop();
//当前节点为叶子节点,返回二叉树的最小深度
if (node->left == nullptr && node->right == nullptr)
return depth;
//当前节点不为叶子节点,优先队列中插入该节点的左、右孩子节点,并记录对应深度
if (node->left != nullptr)
q.emplace(node->left, depth + 1);
if (node->right != nullptr)
q.emplace(node->right, depth + 1);
}
return 0;
}
};