系列文章目录
前言
前些天发现了一个巨牛的人工智能学习网站,通俗易懂,风趣幽默,忍不住分享一下给大家。点击跳转到网站,这篇文章男女通用,看懂了就去分享给你的码吧。
描述
给定一个节点数为 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;
}
};