牛客题霸 [二叉树的之字形层序遍历] C++题解/答案

题目描述

给定一个二叉树，返回该二叉树的之字形层序遍历，（第一层从左向右，下一层从右向左，一直这样交替）
例如：
给定的二叉树是{3,9,20,#,#,15,7},

该二叉树之字形层序遍历的结果是
[
[3],
[20,9],
[15,7]
]

题解：

因为是之字形，所以如果是偶数行就从左向右遍历，如果是奇数行就从右向左遍历，我们可以用一个bool变量flag来标记当前是偶还是奇

代码：

/** * struct TreeNode { * int val; * struct TreeNode *left; * struct TreeNode *right; * }; */

class Solution {
   
public:
    /** * * @param root TreeNode类 * @return int整型vector<vector<>> */
    vector<vector<int> > zigzagLevelOrder(TreeNode* root) {
   
        // write code here
       if(root == nullptr) return {
   };
        vector<vector<int>> res;
        queue<TreeNode*> q;
        q.push(root);
        vector<int> temp;
        bool flag = 0;
        while(q.size())
        {
   
            int n = q.size();
            while(n--)
            {
   
                TreeNode* node = q.front();
                q.pop();
                temp.push_back(node->val);
                if(node->left) q.push(node->left);
                if(node->right) q.push(node->right);
            }
            if(flag) reverse(temp.begin(),temp.end());//翻转
            res.push_back(temp);
            temp.clear();
            flag = !flag;
        }
        return res;
    }
};