【刷题之路】二叉树按层打印

按层打印，换层换行

class TreePrinter {
public:
    vector<vector<int> > printTree(TreeNode* root) {
        // write code here
        TreeNode* last;
        TreeNode* nlast;
        vector<int> temp; //用于保存每一层的值
        queue<TreeNode*> tmp; //用于保存遍历的树节点
        vector<vector<int> > res;
        last=root;
        nlast=root;
        tmp.push(root);
        while(!tmp.empty()){
            last=tmp.front(); //last为队列的头，即为遍历的节点
            tmp.pop();
            temp.push_back(last->val);
            if(last->left) tmp.push(last->left); //将当前遍历的节点的左右子节点放入队列中
            if(last->right) tmp.push(last->right);   
            if(last==nlast){
                nlast=tmp.back(); //当last==nlast，下一层的所有节点必然已经全部放入队列中，于是让nlast=tmp的末尾，即为下一行的最后一个节点
                res.push_back(temp);
                temp.clear();
            }     
        }
        return res;
    }
};