题目描述
从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
时间限制:1秒 空间限制:32768K 热度指数:111808
思路
见代码
/*
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> > ans;
if(pRoot==NULL)
return ans;
queue<TreeNode*> q;
q.push(pRoot);
int cnt=1; //此时访问的结点
int cc=1; //当前层
int next=1;//下一层
vector<int> t;
while(!q.empty())
{
t.push_back(q.front()->val);
if(q.front()->left!=NULL)
{//每一个结点进队,表示下一层结点结尾位置+1
q.push(q.front()->left);
++next;
}
if(q.front()->right!=NULL)
{//每一个结点进队,表示下一层结点结尾位置+1
q.push(q.front()->right);
++next;
}
q.pop();
if(cnt==cc)
{//若当前结点已经到当前层的尾端时,将当前层尾端值更新为下一层尾端
ans.push_back(t);
cc=next;
t.clear();
}
++cnt;
}
return ans;
}
};