【Leetcode】103. 二叉树的锯齿形层次遍历

QUESTION

medium

题目描述

给定一个二叉树，返回其节点值的锯齿形层次遍历（即先从左往右，再从右往左进行下一层遍历，以此类推，层与层之间交替进行）

e.g 给定二叉树: [3,9,20,null,null,15,7]

    3
   / \
  9  20
    /  \
   15   7

返回其层次遍历结果：

[
  [3],
  [20,9],
  [15,7]
]

说明

注意遍历的方向

---

SOLUTION

其实就比普通的层次遍历多考虑奇偶层情况

方法一(双栈)

双栈是比较易懂的方法，注意两个栈压入左右孩子的顺序是不一样的

vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
    vector<vector<int>> res;
    if(!root) return res;
    stack<TreeNode*> sr;
    stack<TreeNode*> sl;
    sr.push(root);
    bool flag = true; //向右
    while(!sr.empty() || !sl.empty()){
        vector<int> row;
        TreeNode* p = NULL;
        if(flag){ //向右遍历层
            while(!sr.empty()){
                p = sr.top();
                row.push_back(p->val);
                if(p->left) sl.push(p->left);
                if(p->right) sl.push(p->right);
                sr.pop();
            }
            flag = false;
        }
        else{ //向左遍历层
            while(!sl.empty()){
                p = sl.top();
                row.push_back(p->val);
                if(p->right) sr.push(p->right);
                if(p->left) sr.push(p->left);
                sl.pop();
            }
            flag = true;
        }
        res.push_back(row);
    }
    return res;
}