给你二叉树的根节点 root ,返回其节点值的 锯齿形层序遍历 。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
class Solution {
public:
vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
queue<TreeNode*> q;
if (root != nullptr)
q.push(root);
bool flag = false;
vector<vector<int>> ans;
while (!q.empty()) {
int size = q.size();
vector<int> v;
for (int i = 0; i < size; i++) {
TreeNode* t = q.front();
q.pop();
v.push_back(t->val);
if (t->left)
q.push(t->left);
if (t->right)
q.push(t->right);
}
if(flag)
reverse(v.begin(),v.end());
ans.emplace_back(v);
flag = !flag;
}
return ans;
}
};
本题思路由【力扣刷题练习】102. 二叉树的层序遍历演变而来
在原有思路以及代码下增加一个布尔型flag变量用以控制遍历顺序,每隔一层调用一次reverse翻转顺序最终实现锯齿形层次遍历。