剑指offer——按之字形打印二叉树

题目描述：
请实现一个函数按照之字形打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右至左的顺序打印，第三行按照从左到右的顺序打印，其他行以此类推。
思路：
设两个栈，s1存放奇数层，s2存放偶数层
遍历s1节点的同时按照左子树、右子树的顺序加入s2，
遍历s2节点的同时按照右子树、左子树的顺序加入s1
代码：
语言C++，已通过牛客网在线测试。

/*
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> >  vec;
        if(pRoot == NULL) return vec;
        stack<TreeNode*> stk1, stk2;
        stk1.push(pRoot);
        vector<int> tmp;
        while(!stk1.empty() || !stk2.empty()){
            while(!stk1.empty()){
                TreeNode* pNode = stk1.top();
                tmp.push_back(pNode->val);
                if(pNode->left) stk2.push(pNode->left);
                if(pNode->right) stk2.push(pNode->right);
                stk1.pop();
            }
            if(tmp.size()){
                vec.push_back(tmp);
                tmp.clear();
            }
            while(!stk2.empty()){
                TreeNode* pNode = stk2.top();
                tmp.push_back(pNode->val);
                if(pNode->right) stk1.push(pNode->right);
                if(pNode->left) stk1.push(pNode->left);
                stk2.pop();
            }
            if(tmp.size()){
                vec.push_back(tmp);
                tmp.clear();
            }
        }
        return vec;  
    } 
};