题目链接:https://leetcode.cn/problems/binary-tree-level-order-traversal/description/
题目大意:给一棵树的根,要求以vector<vector<int>>
形式给出层序遍历结果。
思路:层序遍历并不难,tricky的点在于(如果用c++)做的话,如何让第i
层的结果放进res[i]
里。因为单纯只用一个queue做层序遍历(其实就是BFS)的话,并不能区分哪个节点属于哪一层。
开始时想着一起记录深度,然后放进结果里。但结果数组要开多大呢?即使先开了2000,但有可能有些深度并没有数字,就变成空数组了,这样的结果也是不合法的。
看了题解才发现这个trick:按照轮次来pop queue,可以用这样的循环来保证这轮pop的刚好都是同一层。因为循环是递减的,i
只被初始化一次,即使后面有新的节点被加入了queue,也不影响循环次数(这个次数一开始就被定好了,并且就是这一层的节点数量)。
for (int i = q.size(); i > 0; i--) {
auto parent = q.front();
q.pop();
tmp.push_back(parent->val);
if (parent->left != nullptr)
q.push(parent->left);
if (parent->right != nullptr)
q.push(parent->right);
}
完整代码
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
vector<vector<int>> res;
queue<TreeNode*> q;
if (root != nullptr)
q.push(root);
while (!q.empty()) {
vector<int> tmp;
for (int i = q.size(); i > 0; i--) {
auto parent = q.front();
q.pop();
tmp.push_back(parent->val);
if (parent->left != nullptr)
q.push(parent->left);
if (parent->right != nullptr)
q.push(parent->right);
}
res.push_back(tmp);
}
return res;
}
};