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;}
};