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