二叉树层序遍历

题目描述

给你二叉树的根节点 root ，返回其节点值的 层序遍历 。 （即逐层地，从左到右访问所有节点）。

假设有这样一棵二叉树，那么它经过层序遍历的结果就应该是：

[[3],[9,20],[15,7]]

解法

我们可以用广度优先搜索解决这个问题。

• 按层打印： 题目要求的二叉树的 从上至下 打印（即按层打印），又称为二叉树的 广度优先搜索，通常借助 队列 的先入先出特性来实现。
• 每层打印到一行： 将本层全部节点打印到一行，并将下一层全部节点加入队列，以此类推，即可分为多行打印。

初始化： 打印结果列表 res = [] ，包含根节点的队列 queue = [root] 。

循环： 当队列 queue 为空时跳出。
        新建一个临时列表 tmp ，用于存储当前层打印结果。
        当前层打印循环： 循环次数为当前层节点数（即队列 queue 长度）。
                出队： 队首元素出队，记为 node。
                打印： 将 node.val 添加至 tmp 尾部。
                添加子节点： 若 node 的左（右）子节点不为空，则将左（右）子节点加入队列 queue 
        将当前层结果 tmp 添加入 res 。

返回： 返回打印结果列表 res 即可。

需要注意的是：进行特例处理，当根节点为空，则返回空列表 

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {

        queue<TreeNode*> q;//设置辅助队列

        vector<vector<int>> res;//用来存储最终结果

        if (root != nullptr) q.push(root);
        while (!q.empty()) {

            vector<int> tmp;//每一层遍历以前都初始化tmp
//这里需注意由于在for循环的执行过程中q.size()的值会进行更改，因此q.size()只能作为i的起点而不是终点
            for(int i = q.size(); i > 0; --i) {
                root = q.front();
                q.pop();
                tmp.push_back(root->val);//在出队列的时刻进行打印，出队列的同时左右结点都入队列
                if (root->left != nullptr) q.push(root->left);
                if (root->right != nullptr) q.push(root->right);
            }
            res.push_back(tmp);
        }
        return res;
    }
};