时间限制:1秒
空间限制:32768K
热度指数:56366
算法知识视频讲解
题目描述
从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
class Solution {
public:
vector<vector<int> > Print(TreeNode* pRoot) {
vector<vector<int> > vec;
if (pRoot==NULL)
{
return vec;
}
queue<TreeNode*> q1;
q1.push(pRoot);
while (!q1.empty())
{
int s = q1.size();
vector<int> v;
while (s--)
{
TreeNode *q = q1.front();
q1.pop();
v.push_back(q->val);
if (q->left)
{
q1.push(q->left);
}
if (q->right)
{
q1.push(q->right);
}
}
vec.push_back(v);
}
return vec;
}
};
在二叉树的层次化遍历中一般使用队列进行辅助操作,在层次化的打印中也使用队列,但是不同于层次化遍历的一点是,需要对每个层次的节点数进行统计,然后按行打印,队列中每次循环中放的恰好是这一层元素的个数,然后放在一个一维向量中,循环结束后放到二维向量中,直到队列为空。