牛客网——按之字形顺序打印二叉树（C++）

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

C++

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
class Solution {
public:
    void dfs(TreeNode* node, int k, map<int,vector<int>>& tmp)
    {
        if(node)
        {
            tmp[k].push_back(node->val);
            dfs(node->left,k+1,tmp);
            dfs(node->right,k+1,tmp);
        }
    }
    
    vector<vector<int> > Print(TreeNode* pRoot) 
    {
        map<int,vector<int>> tmp;
        dfs(pRoot,0,tmp);
        vector<vector<int>> ans;
        for(auto& it:tmp)
        {
            if(0==it.first%2)
            {
                ans.push_back(it.second);
            }
            else
            {
                reverse(it.second.begin(),it.second.end());
                ans.push_back(it.second);
            }
        }
        return ans;
    }
    
};