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

牛客题霸 [二叉树的之字形层序遍历] 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;
    }
};