按行打印二叉树

题目描述

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

 

思路：维护两个指针：last和nlast，其中，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) {
        vector<vector<int>> res;
        if(pRoot==NULL)return res;
        vector<int> tmp;
        std::queue<TreeNode *> que;
        que.push(pRoot);
        TreeNode *last=pRoot;
        TreeNode *nlast=pRoot;
        while(que.size())
        {
            TreeNode *p=que.front();
            que.pop();
            tmp.push_back(p->val);
            if(p->left)
            {
                que.push(p->left);
                nlast=p->left;
            }
            if(p->right)
            {
                que.push(p->right);
                nlast=p->right;
            }
            if(p==last)
            {
                res.push_back(tmp);
                tmp.clear();
                last=nlast;
            }
        }
        return res;
    }
};