剑指 Offer 32 - II. 从上到下打印二叉树

剑指 Offer 32 - II. 从上到下打印二叉树

从上到下按层打印二叉树，同一层的节点按从左到右的顺序打印，每一层打印到一行。
例如:
给定二叉树: [3,9,20,null,null,15,7],
返回其层次遍历结果：[ [3],[9,20],[15,7]]
思路：
本题相对于之前的一道题，从上到下打印出二叉树的每个节点，同一层的节点按照从左到右的顺序打印。需将每一层打印到一行 。
先谈论之前的做法：
题目要求的二叉树的 从上至下 打印（即按层打印），又称为二叉树的 广度优先搜索（BFS）。
BFS 通常借助 队列 的先入先出特性来实现。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<int> levelOrder(TreeNode* root) 
    {
        queue<TreeNode*>Queue;
        if(!root) return {};
        vector<int> res;
        Queue.push(root);
        while(!Queue.empty())
        {
            auto tmp = Queue.front();
            res.push_back(tmp->val);
            if(tmp->left)
            {
                Queue.push(tmp->left);
            }
            if(tmp->right)
            {
                Queue.push(tmp->right);
            }
            Queue.pop();
        }
        return res;
    }
    
};

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

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int>> res;
        queue<TreeNode*> Queue;
        if( !root) return {};
        Queue.push(root);
        while(!Queue.empty())
        {
            vector<int>temp;
            for( int i = Queue.size(); i > 0; i--)
            {
                TreeNode *node = Queue.front();
                Queue.pop();
                temp.push_back(node->val);
                if( node->left != nullptr) Queue.push(node->left);
                if( node->right != nullptr) Queue.push(node->right);
            }
            res.push_back(temp);
        }
        return res;
    }
};

参考：
作者：jyd
链接：https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-ii-lcof/solution/mian-shi-ti-32-ii-cong-shang-dao-xia-da-yin-er-c-5/
来源：力扣（LeetCode）