剑指 Offer 32 - III. 从上到下打印二叉树 III

剑指 Offer 32 - III. 从上到下打印二叉树 III

解题思路：
借助辅助队列和栈，使用特殊标记隔离每层，使用变量i记录层数。
借助队列和栈的特点，将奇数层的从队列弹出打印，将偶数层的从栈弹出打印。

1. 将根节点和特殊标记入队，同时记录当前层数为1。

2. 队列和栈同时空跳出循环：
   1）当前层为奇数层，从队列弹出节点，将节点加入volum，同时将非空左右节点入队，若弹出节点为标志则将volum加入res，清空volum，然后层数+1，弹出标致判断队列非空将标志加入队列。
   2）当前层数为偶数层，循环判断队首节点不是特殊标志则将节点非空左右加入队列，同时弹出节点加入栈中，知道特殊标志跳出，然后取出栈顶元素加入volum，若栈空则将volum加入res，清空volum，同时层数+1，弹出队首特殊标志，队列非空将特殊标志加入队列。

3. 返回res。

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        if(root == NULL)
            return {};
        vector<vector<int>> res;
        vector<int> volum;
        queue<TreeNode*> que;
        stack<TreeNode*> sta;
        TreeNode* pMark = NULL;

        que.push(root);
        que.push(pMark);
        int i = 1;
        while(!que.empty() || !sta.empty())
        {
            if(i % 2 != 0)
            {
                if(que.front() != pMark){
                    TreeNode* node = que.front();
                    que.pop();
                    volum.push_back(node->val);
                    if(node->left != NULL)
                        que.push(node->left);
                    if(node->right != NULL)
                        que.push(node->right);
                }
                else{
                    res.push_back(volum);
                    volum.clear();
                    i++;
                    que.pop();
                    if(!que.empty())
                        que.push(pMark);
                }
            }
            else
            {
                while(que.front() != pMark)
                {
                    TreeNode* node = que.front();
                    if(node->left != NULL)
                        que.push(node->left);
                    if(node->right != NULL)
                        que.push(node->right);
                    que.pop();
                    sta.push(node);
                }
                TreeNode* temp = sta.top();
                sta.pop();
                volum.push_back(temp->val);
                if(sta.empty())
                {
                    res.push_back(volum);
                    volum.clear();
                    i++;
                    que.pop();
                    if(!que.empty())
                        que.push(pMark);
                }
            }
        }
        return res;
    }
};