Leetcode 103. 二叉树的锯齿形层次遍历 解题思路及C++实现

解题思路：

因为是逐层的访问，在访问某一层的节点时，需要将其子节点push进栈，所以自然而然地想到用两个栈a和b。因为是锯齿形的访问顺序，所以，如果将根节点当做第一层，奇数层与偶数层的进栈区别在于：奇数层节点的左子节点先进栈，右子节点后进栈，访问偶数层节点时，顺序相反，详细可看下面的代码。下面的程序中，使用了一个flag来判断是奇数层还是偶数层，没循环一层，flag做一次正负反转。

大循环的逻辑是逐层访问，每访问一层，栈a和栈b，必有一个是空的。循环停止条件是两个栈都为空。

 

/**
 * 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 == NULL) return {};
        
        vector<vector<int> > res;
        int flag = 1;
        stack<TreeNode*> a, b;
        a.push(root);
        while(!a.empty() || !b.empty()){
            vector<int> cur;  //存储每一层的访问结果
            while(flag == 1 && !a.empty()){    //奇数层（把根节点当做第1层）
                root = a.top();
                cur.push_back(root->val);
                a.pop();
                if(root->left != NULL) b.push(root->left);
                if(root->right != NULL) b.push(root->right);
            }
            while(flag == -1 && !b.empty()){   //偶数层
                root = b.top();
                cur.push_back(root->val);
                b.pop();
                if(root->right != NULL) a.push(root->right);
                if(root->left != NULL) a.push(root->left);
            }
            flag = -1 * flag;      //每访问一层，反转一下
            res.push_back(cur);    //存到结果容器中
        }
        return res;
    }
};