给定一个二叉树,找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明: 叶子节点是指没有子节点的节点。
示例:
给定二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回它的最小深度 2.
BFS:
class Solution {
public:
int minDepth(TreeNode* root) {
if(!root) return 0;
queue<TreeNode*> q{{root}};
int level = 0;
while(!q.empty())
{
level++;
for(int i = q.size();i > 0;i--)
{
TreeNode* t = q.front();q.pop();
if(!t->left && !t->right)
return level;
if(t->left) q.push(t->left);
if(t->right) q.push(t->right);
}
}
return -1;
}
};
递归:
class Solution {
public:
int minDepth(TreeNode* root) {
if (!root) return 0;
if (!root->left) return 1 + minDepth(root->right);
if (!root->right) return 1 + minDepth(root->left);
return 1 + min(minDepth(root->left), minDepth(root->right));
}
};
DFS:用栈
class Solution {
public:
int minDepth(TreeNode* root) {
if(!root) return 0;
stack<pair<TreeNode*,int>> ss;
ss.push(make_pair(root,1));
int res = INT_MAX;
while(!ss.empty())
{
TreeNode * t = ss.top().first;
int level = ss.top().second;
ss.pop();
if(!t->left && !t->right)
{
res = min(level,res);
}
if(t->left)
ss.push(make_pair(t->left,level + 1));
if(t->right)
ss.push(make_pair(t->right,level + 1));
}
return res;
}
};