[leetcode]剑指 Offer 32 - III. 从上到下打印二叉树 III

请实现一个函数按照之字形顺序打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右到左的顺序打印，第三行再按照从左到右的顺序打印，其他行以此类推。

例如:

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

    3
   / \
  9  20
    /  \
   15   7
返回其层次遍历结果：

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

提示：

节点总数 <= 1000

思路：

基本思路同[leetcode]剑指 Offer 32 - II. 从上到下打印二叉树 II，在每一层遍历结束之后加入结果数组之前加一个判断，定义一个flag标记当前是奇数层还是偶数层，奇数层直接加入，偶数层反转之后加入。

AC代码：（C++）

class Solution {
   public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        deque<TreeNode*> dq;
        vector<vector<int>> ans;
        if (root == nullptr) {
            return ans;
        }
        int flag = -1;
        dq.push_front(root);
        while (!dq.empty()) {
            vector<int> level;
            int len = dq.size();
            flag *= -1;
            for (int i = len; i > 0; i--) {
                TreeNode* temp = dq.front();
                dq.pop_front();
                level.push_back(temp->val);
                if (temp->left) {
                    dq.push_back(temp->left);
                }
                if (temp->right) {
                    dq.push_back(temp->right);
                }
            }
            if (flag == 1) {
                ans.push_back(level);
            } else {
                ans.push_back(vector<int>(level.rbegin(), level.rend()));
            }
        }
        return ans;
    }
};