103. 二叉树的锯齿形层序遍历
程序员文章站
2022-05-20 23:13:16
...
给定一个二叉树,返回其节点值的锯齿形层序遍历。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
例如:
给定二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回锯齿形层序遍历如下:
[
[3],
[20,9],
[15,7]
]
解答
使用双向队列deque
实现,根据奇偶层次不同,左右节点插入队列的先后顺序不同,出栈和入栈的方向也不同:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
vector<vector<int>> result;
deque<TreeNode*> q;
int flag = 1;
if(root)
q.push_back(root);
while(!q.empty()){
int len = q.size();
vector<int> temp;
while(len--){
if(flag){
TreeNode* node = q.front();
q.pop_front();
temp.push_back(node->val);
if(node->left){
q.push_back(node->left);
}
if(node->right){
q.push_back(node->right);
}
}
else{
TreeNode* node = q.back();
q.pop_back();
temp.push_back(node->val);
if(node->right){
q.push_front(node->right);
}
if(node->left){
q.push_front(node->left);
}
}
}
flag ^= 1;
result.push_back(temp);
}
return result;
}
};
当然,最容易想到的思路还是基本的层次遍历,然后对奇数层(从0开始计数)的结果翻转处理,直接使用reverse
效率很低,参考评论区,采用vector::rbegin, vector::end
进行处理:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
vector<vector<int>> result;
queue<TreeNode*> q;
if(root)
q.push(root);
int flag = 0;
while(!q.empty()){
int len = q.size();
vector<int> temp;
while(len--){
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){
result.push_back(vector<int>(temp.rbegin(), temp.rend()));
}
else{
result.push_back(vector<int>(temp.begin(), temp.end()));
}
flag ^= 1;
}
return result;
}
};
上一篇: 最好不要得罪医生
推荐阅读