题目
给定一个二叉树,返回其节点值的锯齿形层序遍历。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
例如:
给定二叉树 [3,9,20,null,null,15,7],
返回锯齿形层序遍历如下:
[
[3],
[20,9],
[15,7]
]
思路
本题是二叉树的层序遍历的变形,区别在于本题是一行从左到右遍历,下一行从右到左遍历,交叉往返的之字形遍历。
可以利用层序遍历,增加一个变量cnt来记录当前的层数(cnt从0开始),将所有的奇数层的节点值进行翻转一下。
c++代码
class Solution {
public:
vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
if(!root)
return {};
vector<vector<int>> res;
queue<TreeNode *> q{{root}};
int cnt = 0;
while(!q.empty()){
vector<int> oneLevel;
for(int i = q.size(); i > 0; i-- ){
TreeNode * t = q.front();
q.pop();
oneLevel.push_back(t->val);
if(t->left)
q.push(t->left);
if(t->right)
q.push(t->right);
}
if(cnt % 2 == 1)
reverse(oneLevel.begin(), oneLevel.end());
res.push_back(oneLevel);
cnt++;
}
return res;
}
};