leetcode#103 二叉树的锯齿形层序遍历
题目:
给定一个二叉树,返回其节点值的锯齿形层序遍历。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
示例:
例如:
给定二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回锯齿形层序遍历如下:
[
[3],
[20,9],
[15,7]
]
思路:
1、双端队列
双端队列支持从队尾进行遍历,而且可以从队头插入。
所以,反着遍历当前层的时候可以选择从队尾遍历,然后从队头插入。
代码:
/**
* 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>> ans;
void bfs(TreeNode *root)
{
int depth = 0;
vector<int> tmp;
deque<TreeNode *> qu;
qu.push_back(root);
while (!qu.empty())
{
int size = qu.size();
if (depth % 2 == 0)
{
for (int i = 0; i < size; ++i)
{
tmp.push_back(qu.front()->val);
if (qu.front()->left)
qu.push_back(qu.front()->left);
if (qu.front()->right)
qu.push_back(qu.front()->right);
qu.pop_front();
}
ans.push_back(tmp);
tmp.clear();
}
else
{
for (int i = size-1; i >=0; --i)
{
tmp.push_back(qu.back()->val);
if (qu.back()->right)
qu.push_front(qu.back()->right);
if (qu.back()->left)
qu.push_front(qu.back()->left);
qu.pop_back();
}
ans.push_back(tmp);
tmp.clear();
}
depth++;
}
return;
}
vector<vector<int>> zigzagLevelOrder(TreeNode *root)
{
if (!root)
return {};
bfs(root);
return ans;
}
};