题目描述
请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
思路:
一个Queue+每一层一个NULL标记 即可。
每当队列头部检测到NULL时,代表上一层的遍历已经结束,此时队列中存放的是下一层的节点,所以要在队列尾部中再添加一个NULL。
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
*/
//每当队列头部检测到NULL时,代表上一层的遍历已经结束,此时队列中存放的是下一层的节点,所以要在队列尾部中再添加一个NULL。
class Solution {
public://每一层遍历完之后,在最后添加一个NULL,标记.
vector<vector<int> > Print(TreeNode* pRoot) {
vector<vector<int>> res;
if(pRoot==NULL) return res;
queue<TreeNode*>q;
q.push(pRoot);
q.push(NULL);//每次用NULL标记每一层是否到达每一层的最右边。
vector<int> vec;
bool flag_reverse=false;
while(!q.empty())
{
TreeNode*pcurrent=q.front();
if(pcurrent==NULL) break; //必须要有,假设只有一个节点[1,NULL],防止非法访问NULL->left
q.pop();
vec.push_back(pcurrent->val);
if(pcurrent->left) q.push(pcurrent->left);
if(pcurrent->right) q.push(pcurrent->right);
if(q.front()==NULL){
if(flag_reverse){
reverse(vec.begin(),vec.end());
flag_reverse=false;
}else
flag_reverse=true;
res.push_back(vec);
vec.clear();
q.pop();
q.push(NULL);//每一层遍历完之后,在最后添加一个NULL,标记.
}
}
return res;
}
};