题目描述
给定一个二叉树,找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明:叶子节点是指没有子节点的节点。
示例:
给定二叉树[3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回它的最小深度 2.
算法
BFS,广度优先搜索
每次把一层节点压入队列,同时判断这些节点中是否含有叶子节点(即左右指针都为空),若有,说明找到了最近的那个叶子节点,返回层数。
C++ 代码
class Solution {
public:
int minDepth(TreeNode* root) {
if(root==NULL) return 0; //判空
queue<pair<TreeNode*,int>> que; //把节点和所在的层数绑定
que.push(make_pair(root,1)); //压入根节点,层数为1
while(1)
{
TreeNode* node=que.front().first; //当前节点
int level=que.front().second; //层数
que.pop();
if (!node->left && !node->right) return level;//遇到叶子节点
if(node->left) //压入左节点
que.push(make_pair(node->left,level+1));
if(node->right)//压入右节点
que.push(make_pair(node->right,level+1));
}
}
};