题解 | #牛群的最短路径#
牛群的最短路径
https://www.nowcoder.com/practice/c07472106bfe430b8e2f55125d817358
考察的知识点:二叉树的最小深度、广度优先搜索;
解答方法分析:
- 首先判断根节点是否为空,若为空直接返回深度0。
- 创建一个队列q,并将根节点root入队。
- 初始化深度depth为0。
- 进入循环,直到队列为空。在每一层的循环开始前,记录当前层的节点个数levelSize,并将depth+1。
- 从队列中取出levelSize个节点,对于每个节点:判断节点是否为叶子节点,即左右子节点都为空,若是则返回当前的深度depth。若节点的左子节点不为空,则将左子节点入队。若节点的右子节点不为空,则将右子节点入队。
- 如果循环结束后仍然没有返回深度值,说明二叉树不为空,但没有叶子节点,这时可以返回最外层的深度depth作为最小深度。
- 结束循环后,返回最小深度depth。
所用编程语言:C++;
完整编程代码:↓
/**
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* };
*/
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param root TreeNode类
* @return int整型
*/
int minDepth(TreeNode* root) {
if (root == nullptr) {
return 0;
}
queue<TreeNode*> q;
int depth = 0;
q.push(root);
while (!q.empty()) {
int levelSize = q.size();
depth++;
for (int i = 0; i < levelSize; i++) {
TreeNode* node = q.front();
q.pop();
if (node->left == nullptr && node->right == nullptr) {
return depth;
}
if (node->left != nullptr) {
q.push(node->left);
}
if (node->right != nullptr) {
q.push(node->right);
}
}
}
return depth;
}
};
查看11道真题和解析
