题解 | #牛群分层排列#
牛群分层排列
https://www.nowcoder.com/practice/7e98027a60424c9ba88d9c4c0506ede4
考察的知识点:二叉树的层次遍历;
解答方法分析:
- 首先判断根节点是否为空,如果为空,则直接返回一个空的字符串向量。
- 创建一个字符串向量
res来保存结果,创建一个队列queue来存储待遍历的节点。 - 将根节点入队。
- 循环直到队列为空:创建一个空字符串 sb 来存储当前层的节点值,获取队列当前层节点的数量 count,循环 count 次,依次处理当前层的节点:从队列中取出一个节点 node,将节点的值转化为字符串并追加到 sb 中,如果节点有左子节点,则将左子节点入队。如果节点有右子节点,则将右子节点入队。将 sb 到结果向量 res 中。
- 返回结果向量
res。
所用编程语言: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 string字符串vector
*/
vector<string> levelOrder(TreeNode* root) {
if (root == nullptr) {
return {};
}
vector<string> res;
queue<TreeNode*> queue;
queue.push(root);
while (!queue.empty()) {
string sb = "";
int count = queue.size();
while (count-- > 0) {
TreeNode* node = queue.front();
queue.pop();
sb += to_string(node->val);
if (node->left != nullptr) {
queue.push(node->left);
}
if (node->right != nullptr) {
queue.push(node->right);
}
}
res.push_back(sb);
}
return res;
}
};
查看9道真题和解析