NC 把二叉树打印成多行

系列文章目录

---

---

前言

前些天发现了一个巨牛的人工智能学习网站，通俗易懂，风趣幽默，忍不住分享一下给大家。点击跳转到网站，这篇文章男女通用，看懂了就去分享给你的码吧。

---

描述
给定一个节点数为 n 二叉树，要求从上到下按层打印二叉树的 val 值，同一层结点从左至右输出，每一层输出一行，将输出的结果存放到一个二维数组中返回。
例如：
给定的二叉树是{1,2,3,#,#,4,5}

该二叉树多行打印层序遍历的结果是
[
[1],
[2,3],
[4,5]
]
输入描述：
给定一个二叉树的根节点

/*
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) {
            vector<vector<int> > res;
            if (!pRoot)
                return res;
            queue<TreeNode*> q; // 定义队列
            q.push(pRoot); // 根结点入队
            while (!q.empty()) {
                int size = q.size();
                vector<int> arr;
                for (int i = 0; i < size; i++) {
                    TreeNode* p = q.front();
                    q.pop();
                    if (p->left) q.push(p->left); // 左孩子入队
                    if (p->right) q.push(p->right); // 右孩子入队
                    if (!p) continue;
                    arr.push_back(p->val);
                }
                res.push_back(arr);
                arr.clear();
            }
            return res;
        }
};