按之字形顺序打印二叉树

题目描述

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

思路

层序遍历二叉树只需要一个队列即可。而按照题意，奇数从左到右，偶数层从右到左，则利用一个队列和一个栈来实现。队列先进先出，栈先进后出，所以队列保存奇数层结点，栈保存偶数层结点。由于偶数层是从右到左，所以栈直接从左到右即可。而奇数层需要从右到左，所以将奇数层结点按照从右到左插入到队列的头部。

代码

/*
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>> res;
        if (pRoot == nullptr) return res;
        bool flag = true;
        stack<TreeNode*> s;
        deque<TreeNode*> q;
        q.push_back(pRoot);
        while (!q.empty() || !s.empty()){
            vector<int> tmp;
            if (flag){
                int size = q.size();
                for (int i = 0; i < size; ++i){
                    auto cur = q.front();
                    tmp.emplace_back(cur->val);
                    q.pop_front();
                    if (cur->left) s.emplace(cur->left);
                    if (cur->right) s.emplace(cur->right);
                }
            }
            else{
                int size = s.size();
                for (int i = 0; i < size; ++i){
                    auto cur = s.top();
                    tmp.emplace_back(cur->val);
                    s.pop();
                    if (cur->right) q.push_front(cur->right);
                    if (cur->left) q.push_front(cur->left);
                }
            }
            res.emplace_back(tmp);
            flag = !flag;
        }
        return res;
    }
};