Leetcode 103. Binary Tree Zigzag Level Order Traversal 树的之字形打印 栈实现

树的之字形打印

题目链接

题目描述

给出一个二叉树，将其按照层次遍历打印，但是奇数行正序打印，偶数行逆序打印。

解题思想

树的层次遍历，需要采用队列实现。这里与层次遍历时不同，因为对于偶数行，我们需要逆序打印，想到可以利用一个栈来帮助打印。对于偶数行的数，不直接打印，而是存在一个栈里，等到了这一行的末尾再进行打印。

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
        if (!root) return {};
        list<TreeNode*> l;
        TreeNode* last = root;
        l.push_back(root);
        stack<int> s;
        vector<vector<int>> result;
        std::vector<int> v;
        bool isodd = true;
        while(!l.empty()){
            TreeNode* t = l.front();
            l.pop_front();
            if (isodd){
                v.push_back(t -> val);
            }
            else{
                s.push(t -> val);
            }
            if (t -> left)
                l.push_back(t -> left);
            if (t -> right)
                l.push_back(t -> right);
            if (t == last){
                last = l.back();
                while(!s.empty()){
                    v.push_back(s.top());
                    s.pop();
                }
                result.push_back(v);
                v.clear();
                isodd ^= 1;
            }
        }
        return result;
    }
};