103. 二叉樹的鋸齒形層序遍歷

題目連結

來源：力扣（LeetCode）
連結：https://leetcode-cn.com/problems/binary-tree-zigzag-level-order-traversal/
著作權歸領釦網路所有。商業轉載請聯絡官方授權，非商業轉載請註明出處。

題目描述

給定一個二叉樹，返回其節點值的鋸齒形層序遍歷。（即先從左往右，再從右往左進行下一層遍歷，以此類推，層與層之間交替進行）。

例如：
給定二叉樹 [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7

返回鋸齒形層序遍歷如下：

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

題目分析

1. 根據題目描述鋸齒形層次遍歷二叉樹
2. 按廣度優先搜尋遍歷二叉樹，反轉其中第2、4、6、...層節點序列

程式碼

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
        vector<vector<int>> res;
        if (!root) return res;
        queue<TreeNode*> queNode;
        vector<int> nodeVal;
        int queNodeLen;
        queNode.push(root);
        int index = 0;
        while (!queNode.empty()) {
            queNodeLen = queNode.size();
            nodeVal.clear();
            while (queNodeLen--) {
                nodeVal.push_back(queNode.front()->val);
                if (queNode.front()->left) queNode.push(queNode.front()->left);
                if (queNode.front()->right) queNode.push(queNode.front()->right);
                queNode.pop();
            }
            if (index % 2 == 1) reverse(nodeVal.begin(), nodeVal.end());
            ++index;
            res.push_back(nodeVal);
        }
        return res;
    }
};