题目描述
从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
用last记录上一层最后一个
用nlast记录下一层最后一个;
当打印到last的时候,令last=nlast,保存一层即可。
/*
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) {
if(pRoot==NULL)
return vector<vector<int>>{};
vector<vector<int> > res;
vector<int> vec;
queue<TreeNode*> que;
que.push(pRoot);
TreeNode* last=pRoot;
TreeNode* nlast=pRoot;
while(!que.empty())
{
TreeNode* temp=que.front();
que.pop();
if(temp->left){
que.push(temp->left);
nlast=temp->left;
}
if(temp->right){
que.push(temp->right);
nlast=temp->right;
}
vec.push_back(temp->val);
if(temp==last)
{
res.push_back(vec);
last=nlast;
vec.clear();
}
}
return res;
}
};