JZ77 按之字形顺序打印二叉树

描述

给定一个二叉树，返回该二叉树的之字形层序遍历，（第一层从左向右，下一层从右向左，一直这样交替）

数据范围：0 \le n \le 15000≤n≤1500,树上每个节点的val满足 |val| <= 1500∣val∣<=1500
要求：空间复杂度：O(n)O(n)，时间复杂度：O(n)O(n)
例如：
给定的二叉树是{1,2,3,#,#,4,5}

该二叉树之字形层序遍历的结果是
[
[1],
[3,2],
[4,5]
]
示例1
输入：
{1,2,3,#,#,4,5}
返回值：
[[1],[3,2],[4,5]]
说明：
如题面解释，第一层是根节点，从左到右打印结果，第二层从右到左，第三层从左到右。

示例2
输入：
{8,6,10,5,7,9,11}
返回值：
[[8],[10,6],[5,7,9,11]]
复制
示例3
输入：
{1,2,3,4,5}
返回值：
[[1],[3,2],[4,5]]

代码

/*
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 == NULL) return res;
        //使用栈的特性
        stack<TreeNode *> s1, s2;
        s1.push(pRoot);
        int k=2;
        while(!s1.empty()) {
            vector<int> line;
            while(!s1.empty()) {
                TreeNode *cur = s1.top();
                s1.pop();
                line.push_back(cur->val);
                if(k%2 == 0){
                    if(cur->left) s2.push(cur->left);
                    if(cur->right) s2.push(cur->right);
                }
                else {
                    if(cur->right) s2.push(cur->right);
                    if(cur->left) s2.push(cur->left);
                }
            }
            k++;
            res.push_back(line);
            swap(s1,s2);
        }
        return res;
    }
    
};