题目链接: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;}
};