给定一个二叉树,返回其按层次遍历的节点值。 (即逐层地,从左到右访问所有节点)。
例如:
给定二叉树: [3,9,20,null,null,15,7]
,
3 / \ 9 20 / \ 15 7
返回其层次遍历结果:
[ [3], [9,20], [15,7] ]
思路:
二叉树的层序遍历是基础,这里参考之前一道题的discuss部分,用一个queue即可实现,不用两个queue(每次从一个出去填充另外一个),通过判断每次queue的size达到只在内部循环size次来避免访问到子节点。
代码如下:
vector<vector<int>> levelOrder(TreeNode* root) {
vector<vector<int>> res;
if (!root) {
return res;
}
queue<TreeNode*> queue;
queue.push(root);
while (!queue.empty()) {
int q = queue.size();
vector<int> tmp;
for (int i = 0; i < q; i++) {
tmp.push_back(queue.front()->val);
if (queue.front()->left) {
queue.push(queue.front()->left);
}
if (queue.front()->right) {
queue.push(queue.front()->right);
}
queue.pop();
}
res.push_back(tmp);
}
return res;
}