按之字形顺序打印二叉树

题目描述
请实现一个函数按照之字形打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右至左的顺序打印，第三行按照从左到右的顺序打印，其他行以此类推。

class Solution {
public:
    vector<vector<int>> Print(TreeNode* pRoot) {
        vector<vector<int>> ans;
        queue<TreeNode*> q;
        int dir=0;
        if(pRoot){
            q.push(pRoot);
            while(!q.empty()){
                int cnt=q.size();
                vector<int> tmp;
                for(int i=0;i<cnt;++i){
                    tmp.push_back(q.front()->val);
                    if(q.front()->left)
                        q.push(q.front()->left);
                    if(q.front()->right)
                        q.push(q.front()->right);
                    q.pop();
                }
                if(dir%2)
                    reverse(tmp.begin(),tmp.end());
                dir++;
                ans.push_back(tmp);
            }
        }
        return ans;
    }
};