利用queue
先进先出的特性进行处理,利用queue.size()
来识别元素是否在二叉树的同一层,同时要注意不能直接i < q.size()
来判断,因为q.size()
是不断变化的。
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode *root) {
queue<TreeNode *> q;
vector<vector<int>> res;
int depth = 0;
if (root != nullptr)
q.push(root);
while (!q.empty()) {
vector<int> temp;
// 如何识别节点在同一层?利用queue的size()来进行处理。
int size = q.size();
// 不能直接i<q.size(),因为循环中, q.size()是变化的
for (int i = 0; i < size; i++) {
TreeNode *node = q.front();
q.pop();
temp.push_back(node->val);
if (node->left != nullptr)
q.push(node->left);
if (node->right != nullptr)
q.push(node->right);
}
res.push_back(temp);
}
// std::reverse(res.begin(), res.end());
return res;
}
};
标签:node,binary,level,res,queue,二叉树,push,size
From: https://www.cnblogs.com/zwyyy456/p/16608657.html