leecode.103. 二叉树的锯齿形层序遍历

题目

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

例如：
给定二叉树 [3,9,20,null,null,15,7],

返回锯齿形层序遍历如下：

思路分析

• 每次输出队列头的元素的数值
• 遍历入队的时候，如果当前是奇数层，遍历顺序应该是从左到右，同时将左右的孩子入队；如果当前是偶数层，遍历顺序应该是从右到左，此时将孩子从队头入队。

代码

/**
 * 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) {
        vector<vector<int> > ans;
        vector<int> tmp;
        deque<TreeNode*> q;
        if(!root) return ans;
        bool flag = true;
        q.push_back(root);
        while(!q.empty()){
            int l = q.size();
            for(int i = 0;i < l;i++){
                TreeNode* cur;
                if(!flag){
                    //偶数层，从右到左，弹出队尾元素。下一层是从左到右，右左孩子入队首
                    cur = q.back();
                    q.pop_back();
                    if(cur->right) q.push_front(cur->right);
                    if(cur->left) q.push_front(cur->left);
                }else{
                    //奇数层，从左到右，弹出队首元素，左右孩子入队列尾
                    cur = q.front();
                    q.pop_front();
                    if(cur->left) q.push_back(cur->left);
                    if(cur->right) q.push_back(cur->right);
                }
                tmp.push_back(cur->val);
            }
            ans.push_back(move(tmp));
            flag = !flag;
        }
        return ans;
    }
};