剑指offer:二叉树打印成多行

描述

给定一个节点数为 n 二叉树，要求从上到下按层打印二叉树的 val 值，同一层结点从左至右输出，每一层输出一行，将输出的结果存放到一个二维数组中返回。

例如：
给定的二叉树是{1,2,3,#,#,4,5}

该二叉树多行打印层序遍历的结果是

[

[1],

[2,3],

[4,5]

]

数据范围：二叉树的节点数 0 \le n \le 10000≤n≤1000，0 \le val \le 10000≤val≤1000

要求：空间复杂度 O(n)O(n)，时间复杂度 O(n)O(n)

输入描述：

给定一个二叉树的根节点

示例1

输入：

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

复制返回值：

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

思路:

1.和层次遍历思路一致,使用队列存储节点

2.需要知道当前层的节点数量

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
class Solution {
public:
        vector<vector<int> > Print(TreeNode* pRoot) {
            if(!pRoot) return {};
            queue<TreeNode*> q;
            q.push(pRoot);
            vector<vector<int> > res;
            while(!q.empty()){
                vector<int> ans;
                int size=q.size(); //当前层的节点数量
                for(int i=0;i<size;i++){
                    TreeNode* node=q.front();
                    q.pop();
                    ans.push_back(node->val);
                    if(node->left)q.push(node->left);
                    if(node->right)q.push(node->right);
                }
                res.push_back(ans);
            }
            return res;
        }
    
};