剑指offer——JZ78 把二叉树打印成多行 解题思路与具体代码【C++】

一、题目描述与要求

把二叉树打印成多行_牛客题霸_牛客网 (nowcoder.com)

题目描述

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

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

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

[

[1],

[2,3],

[4,5]

]

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

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

输入描述：

给定一个二叉树的根节点

示例

示例1：

输入：{1,2,3,#,#,4,5}

返回值：[[1],[2,3],[4,5]]

示例2：

输入：{8,6,10,5,7,9,11}

返回值：[[8],[6,10],[5,7,9,11]]

示例3：

输入：{1,2,3,4,5}

返回值：[[1],[2,3],[4,5]]

示例4：

输入：{}

返回值：[]

---

二、解题思路

根据题目要求把二叉树打印成多行，其实就是对二叉树进行层次遍历，并且把每一层的结果分别存入vector，然后进行返回。因此我们可以利用队列来实现二叉树的层次遍历。

首先判断是否为空树，是的话返回空的vector；

定义队列并且将根结点入队，同时定义辅助指针p用于获取每次弹出的结点；

当栈不为空时，根据栈内目前的大小（也就是每层的结点数）进行for循环，将队列首元素弹出，并将其压入vector中，然后判断其左右子树是否为空，非空则入队；

结束每一层遍历后将row压入v中，最后返回v。

---

三、具体代码

#include <queue>
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param pRoot TreeNode类 
     * @return int整型vector<vector<>>
     */
    
    vector<vector<int> > Print(TreeNode* pRoot) {
        vector<vector<int>> v;//最后返回结果
        //如果是空树，则直接返回空的vector
        if(pRoot==nullptr)  return v;
        //利用队列实现层次遍历
        queue<TreeNode*> q;
        TreeNode* p;
        q.push(pRoot);//压入根结点
        while(!q.empty()){//栈不为空时进行
            vector<int> row;//存储每一行的结果
            int n=q.size();//每层结点多少个，队列大小就是多少
            for(int i=0;i<n;i++){
                p=q.front();
                q.pop();//弹出
                row.push_back(p->val);
                if(p->left) q.push(p->left);//左子树不为空
                if(p->right)q.push(p->right);//右子树不为空
            }
            v.push_back(row);
        }
        return v;
    }
};