题目描述:
给定一个二叉树,找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明: 叶子节点是指没有子节点的节点。
示例:
给定二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回它的最小深度 2.
算法:递归
题目有个陷阱,最小深度的定义是到叶子节点,所以,如果一个二叉树只有右子树,那么最小深度是右子树的深度。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
int minDepth(TreeNode* root) {
if(root==NULL)
return 0;
int left = minDepth(root->left);
int right = minDepth(root->right);
if(left!=0 && right !=0)
return min(left, right) + 1;
else
return max(left, right) + 1;
}
};
算法:BFS
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
int minDepth(TreeNode* root) {
if(root==NULL)
return 0;
queue<pair<TreeNode*, int>>q;
q.push(make_pair(root, 1));
int level = 1; //当前层数
while(!q.empty())
{
level = q.front().second;
TreeNode* p = q.front().first;
q.pop();
if(p->left==NULL && p->right==NULL) //遇到叶子节点,就返回当前层数
return level;
if(p->left)
q.push(make_pair(p->left, level+1));
if(p->right)
q.push(make_pair(p->right, level+1));
}
return level;
}
};