按层遍历二叉树

有一棵二叉树，请设计一个算法，按照层次打印这棵二叉树。
给定二叉树的根结点root，请返回打印结果，结果按照每一层一个数组进行储存，所有数组的顺序按照层数从上往下，且每一层的数组内元素按照从左往右排列。保证结点数小于等于500。

具体要求如图所示：

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};*/

class TreePrinter {
public:
    vector<vector<int> > printTree(TreeNode* root) {
        // write code here

        TreeNode *last = root;//last表示正在打印的当前行的最右节点
        TreeNode *nlast = root;//表示下一行的最右节点
        TreeNode *tmp = NULL;
        queue<TreeNode*> q;

        vector<vector<int> > ans;
        vector<int> ret;

        if(root)
            q.push(root);

        while(!q.empty()){
            tmp = q.front();
            q.pop();
            ret.push_back(tmp->val);

            if(tmp->left){
                q.push(tmp->left);
                nlast = tmp->left;
            }
            if(tmp->right){
                q.push(tmp->right);
                nlast = tmp->right;
            }
            //当当前节点等于当前行的最右节点时，更新last
            //把当前行的所有数据放到二维数组ans中，并且清空存储每行数据的数组ret
            if(tmp == last){
               last = nlast;
               ans.push_back(ret);
               ret.clear();
            }
        }
        return ans;
    }
};