1027-二叉树的层次遍历

原题:牛客
题目描述
给定一个二叉树,返回该二叉树层序遍历的结果,(从左到右,一层一层地遍历)
例如:
给定的二叉树是{3,9,20,#,#,15,7},
在这里插入图片描述

该二叉树层序遍历的结果是
[
[3],
[9,20],
[15,7]
]

示例1
输入

{1,2}

返回值

[[1],[2]]

示例2
输入

{1,2,3,4,#,#,5}

返回值

[[1],[2,3],[4,5]]

说明:本题目包含复杂数据结构TreeNode,点此查看相关信息

思想:
本例考量二叉树的层次遍历,首先题目给出了该二叉树的结构,是动态二叉树;
通过本题的输出样例可知,每一层的输出结果单独放在了一个数组当中,且根据题目给出的提示,使用变长数组存放每一层的遍历结果较为合适,并且最后的结果也说明,最后的结果是存放在一个二维数组中的;
因此,创建一个变长的二维数组ans存放最终的结果,用一个一维变长数组res存放每一层的遍历结果;
二叉树的层次遍历,首先要将根节点入队列,由于给出的是动态二叉树,因此队列中存放的是一个结构体,当队列中没有元素的时候,直接返回结果(必须注意这一点!)为了确保当前这一层已经遍历结束,使用size标记当前队列里结构体的元素个数;
队列当中,遵循先入先出的原则,因此,每次入队一个结构体,之后遍历这个结构体的左右子树的结构体,如果不是空则要一直往队列里面加,每次遍历到一个结点,便输出该结点的值;
代码:

/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 * };
 */
#include <queue>
class Solution {
public:
    /**
     * 
     * @param root TreeNode类 
     * @return int整型vector<vector<>>
     */
    vector<vector<int> > levelOrder(TreeNode* root) {
        // write code here
        vector<vector<int> > ans;                            //存放最终结果
        if(!root)
            return ans;
        queue<TreeNode*> q;                                  //队列存放结构体
        q.push(root);                                        //将根节点入队
        while(!q.empty())                                    //当队列不为空时
        {
            int size = q.size();                             //获取当前队列中的内容多少
            vector<int> res;                                 //该数组用来存放当前队列的遍历结果;
            while(size--)
            {
                TreeNode * font = q.front();                 //获取当前队列的头结点
                q.pop();                                     //该结点出队
                res.push_back(font->val);                    //将该结点的值压入存放当前队列的数组
                if(font->left != NULL) q.push(font->left);    //递归遍历左子树
                if(font->right != NULL) q.push(font->right);  //递归遍历右子树
            }
            ans.push_back(res);                               //存放最终结果
        }
        return ans;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值