题目描述
给定一个二叉树,返回其节点值的锯齿形层次遍历。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
例如:
给定二叉树 [3,9,20,null,null,15,7]
,
3 / \ 9 20 / \ 15 7
返回锯齿形层次遍历如下:
[ [3], [20,9], [15,7] ]
问题分析
此题要用两个栈来实现。当一个栈被一直弹出节点时,另一个栈一直在压入节点,两个栈轮流换岗,直到两个栈都为空时结束树的创建并返回。比如当前层是奇数层,栈q1在一个一个弹出这一层的节点,每弹出一个节点,就加入到数组out中,然后由于是奇数层,所以先将这个节点的左孩子压入栈q2,然后将右孩子压入栈q2。当这层的节点都从栈q1弹出后,栈q1就为空了。接下来进入下一轮循环,将level加1,使当前层成为偶数层。然后栈q2一个一个弹出这一层的节点,每弹出一个节点,就加入到数组out中,然后由于是偶数层,所以先将这个节点的右孩子压入栈q1,然后将左孩子压入栈q1。当这层的节点都从栈q2弹出后,栈q2就为空了。如此以上循环,直到栈q1和栈q2全都为空,说明已经遍历结束了,返回ans。
代码实现
/**
* 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) {
if(!root)
return {};
vector<vector<int>> ans;
stack<TreeNode*> q1, q2;
q1.push(root);
int level = 0;
while(!q1.empty() || !q2.empty()){
level++;
vector<int> out;
if(level % 2 == 1){
for(int i = q1.size(); i > 0; i--){
TreeNode* temp = q1.top();
q1.pop();
out.push_back(temp->val);
if(temp->left)
q2.push(temp->left);
if(temp->right)
q2.push(temp->right);
}
}
else{
for(int i = q2.size(); i > 0; i--){
TreeNode* temp = q2.top();
q2.pop();
out.push_back(temp->val);
if(temp->right)
q1.push(temp->right);
if(temp->left)
q1.push(temp->left);
}
}
ans.push_back(out);
}
return ans;
}
};