按层打印二叉树

题目描述
从上到下按层打印二叉树，同一层结点从左至右输出。每一层输出一行。

用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;
        }

};