LeetCode | Binary Tree Level Order Traversal I,II

100 篇文章 0 订阅
28 篇文章 0 订阅

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;
    } 
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值