Given a binary tree, return the level order traversal of its nodes’
values. (ie, from left to right, level by level).For example: Given binary tree [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
return its level order traversal as: [ [3], [9,20], [15,7] ]
对于二叉树的层次遍历,一般使用BFS的办法,利用队列模拟
不过这里的输出,就要求知道当前属于哪一列
这就很尴尬了..
我写了一个在取了元素之后判断Q.empty()的
其实这个是错误的,因为下一个头元素将它的子元素同时也推入队列,所以除了第一行是正确的结果之外别的都是错误的。。。
这个双队列简直治愈~
利用双队列,巧妙的将取出来的元素放到另一个队列里,这样就可以保证这两个队列的内容分开了。
当把元素取完的时候,Q=Qnext;将另一队列的指针赋值给当前队列,
这样一来就完美地完成了两个队列的数据交换。
class Solution{
public:
vector<vector<int> > levelOrder(TreeNode* root) {
vector<vector<int> > result;
if(!root) return result;
queue<TreeNode*> Q;
Q.push(root);
int count=0;
while(!Q.empty()){
//双队列
queue<TreeNode*> Qnext;
vector<int> row;
while(!Q.empty()){
TreeNode* node=Q.front();
Q.pop();
row.push_back(node->val);
//错误示例~没有考虑到后面继续推入会有问题
// //说明弹出的是最后一个
// if(Q.empty()){
// count++;
// result.push_back(row);
// row.erase(row.begin(),row.end());
// }
if(node->left) Qnext.push(node->left);
if(node->right) Qnext.push(node->right);
}
result.push_back(row);
Q=Qnext;
}
//II中只需要添加这一句
//reverse(result.begin(),result.end());
return result;
}
};