【剑指offer】按之字形顺序打印二叉树

题目描述

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

思路：采用两个栈，交换进行存储，$0$栈存储奇数层，$1$栈存储偶数层；当某个栈在打印该层的节点时，将下一层的节点存入另一个栈中，当为$0$栈时，为奇数层，顺序将下一层的节点存入$1$栈中，这样当访问$1$栈时便是先访问右节点后访问左节点；当为$1$栈时，为偶数层，先将右节点存入栈$0$，后将左节点存入栈$0$，当访问$0$栈时，便是先访问左节点后访问右节点；当该层的节点访问结束后，将数据存入$vector$，交换$current$与$next$的值，访问下一层

/*
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> > ans;
    vector<int> vec;
    vector<vector<int> > Print(TreeNode* pRoot) {
        if (pRoot == nullptr)
            return ans;
        //两个栈存放不同层的数据
        stack<TreeNode*> level[2];
        int current = 0, next = 1;
        
        level[current].push(pRoot);
        //访问某一层的节点的同时，将其下一层的节点存放其中；针对不同层不同的放置顺序
        while (!level[0].empty() || !level[1].empty()) {
            TreeNode* pNode = level[current].top();
            level[current].pop();
            
            vec.push_back(pNode->val);
            //current为0时代表奇数层，为1时代表偶数层
            if (current == 0) {
                if (pNode->left != nullptr) {
                    level[next].push(pNode->left);
                } 
                if (pNode->right != nullptr) {
                    level[next].push(pNode->right);
                }
            } else {
                if (pNode->right != nullptr) {
                    level[next].push(pNode->right);
                }
                if (pNode->left != nullptr) {
                    level[next].push(pNode->left);
                }
            }
            if (level[current].empty()) {
                ans.push_back(vec);
                vec.clear();
                current = 1 - current;
                next = 1 - next;
            }
        }
        return ans;
    }
};