给定一个二叉树,找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明: 叶子节点是指没有子节点的节点。
示例:
给定二叉树 [3,9,20,null,null,15,7]
,
3 / \ 9 20 / \ 15 7
返回它的最小深度 2.
题目标签:Tree / Depth-first Search / Breadth-first Search
题目链接:LeetCode / LeetCode中国
由于求的是最小深度,因此考虑层次遍历,注意层次遍历的写法。
Language | Runtime | Memory |
---|---|---|
cpp | 4 ms | N/A |
/**
* 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){
return 0;
}
queue<TreeNode*> seq;
unordered_set<TreeNode*> vt;
vt.insert(root);
seq.push(root);
int res = 0;
while(!seq.empty()){
int cnt = seq.size();
res++;
// 层次遍历。注意写法!
while(cnt--){
TreeNode* node = seq.front();
seq.pop();
if(!node->left && !node->right){
return res; // 当是叶子节点时,立即返回
}
if(node->left && !vt.count(node->left)){
vt.insert(node->left);
seq.push(node->left);
}
if(node->right && !vt.count(node->right)){
vt.insert(node->right);
seq.push(node->right);
}
}
}
}
};