题目描述:
给定一个二叉树,返回其节点值的锯齿形层序遍历。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
例如:
给定二叉树 [3,9,20,null,null,15,7],
3
/
9 20
/
15 7
返回锯齿形层序遍历如下:
[
[3],
[20,9],
[15,7]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-zigzag-level-order-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
//核心思想,用一个变量记录当前层是奇数还是偶数,奇数从左到右,偶数从右到左
//使用普通队列进行遍历,再使用一个双端队列存储一整行的数据
class Solution {
public:
vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
vector<vector<int>> ans;
if(!root) return ans;
queue<TreeNode*> qnode;
bool orderByLeft = true;//奇数层
qnode.push(root);//层次遍历先看后进
while(!qnode.empty()){//区别于传统的层次遍历,这里每次while遍历一整层
int levelSize = qnode.size();
deque<int> level;
for(int i = 0;i < levelSize; i++){
auto curNode = qnode.front();
qnode.pop();
//判断奇数还是偶数层,对应不同的方向插入双端队列,本题逻辑
if(orderByLeft){
level.push_back(curNode->val);
}
else{
level.push_front(curNode->val);
}
//普通层次遍历
if(curNode->left != NULL)
qnode.push(curNode->left);
if(curNode->right != NULL)
qnode.push(curNode->right);
}
orderByLeft = !orderByLeft;//取反,为下次遍历做准备
vector<int> curlevel{level.begin(),level.end()};
ans.push_back(curlevel);
}
return ans;
}
};