题解 | #牛群分层排列#
牛群分层排列
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; } };