题目描述
给定一个二叉树,返回该二叉树的之字形层序遍历,(第一层从左向右,下一层从右向左,一直这样交替)
例如:
给定的二叉树是{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;
}
};