102.二叉树的层序遍历

标题

链接:leetcode链接
在这里插入图片描述

思路分析

二叉树的单纯层序遍历还是很简单的,但是这个题,要求我们将返回值按层写到一个二维数组中,该如何实现呢?
单纯层序遍历使用队列就可以了,每次出队列的时候,进行输出,并将该元素的left和right入队列即可。

在这个题目中,问题在于,如何分辨队列中哪些元素在同一层。
这里有两个方法:

法一:(好想)
再来一个queue存每一个元素的层数。

法二:(好用)
经过观察,可以发现,每一次删完一层的元素之后,队列里面剩余的元素个数就是下一层的元素个数。
我们可以设置一个变量来记录每一层的元素个数。

解决完关键问题之后就比较简单了,套上一般层序遍历的代码来写即可。

注意:这个题目还比较坑,需要考虑空树,需要特判一下。

代码

        
        queue<TreeNode*> q;
        vector<vector<int>> vv;

        if(root == nullptr) return vv;
        int levelSize = 1;
        q.push(root);

        while(!q.empty())
        {
            levelSize = q.size();

            vector<int> v;
            while(levelSize--)
            {
                TreeNode* front = q.front();
                
                v.push_back(front->val);
                q.pop();
                if(front->left) q.push(front->left);
                if(front->right) q.push(front->right);

            }

            vv.push_back(v);

        }

        return vv;
    }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值