题目描述
从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
/*
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> > vv;
vector<int> ve;
queue<TreeNode*> que;
queue<int> dque;
int depth=0;
que.push(pRoot);
dque.push(depth);
TreeNode *node;
while(!que.empty())
{
node=que.front();
if(depth!=dque.front())
{
vv.push_back(ve);
ve.clear();
depth=dque.front();
}
que.pop();
dque.pop();
if(node)
{
que.push(node->left);
que.push(node->right);
dque.push(depth+1);
dque.push(depth+1);
ve.push_back(node->val);
}
}
return vv;
}
};
链接:https://www.nowcoder.com/questionTerminal/445c44d982d04483b04a54f298796288
来源:牛客网
//用两个队列实现按层打印树
vector<vector<int> > Print2(TreeNode* pRoot)
{
vector<vector<int> > res;
if(pRoot==nullptr) return res;
queue<TreeNode*> q1,q2;//用队列缓存树节点的print顺序
q1.push(pRoot);
while(!q1.empty())
{
vector<int> tmp;
while(!q1.empty())
{
tmp.push_back(q1.front()->val);//依次取出栈中元素的值
if(q1.front()->left) q2.push(q1.front()->left);//将栈顶元素按从左至右顺序压入队列
if(q1.front()->right) q2.push(q1.front()->right);
q1.pop();//删除栈顶元素
}
while(!q2.empty()) //将队列中的元素依次入栈
{
q1.push(q2.front());
q2.pop();
}
res.push_back(tmp);
}
return res;
}