1,题目要求
请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
2,题目思路
对于这道题,要求将一颗二叉树按照之字形层次输出。
所谓的之字形,也称为Z字形输出,也是按照层次进行遍历,只不过按照从左往右、从右往左、再从左往右的顺序进行遍历。
经过观察之后我们发现,奇数层都是从左往右,偶数层都是从右往左,因此,我们只需要按照常规层次遍历的方法进行遍历,然后根据当前层的奇偶情况,判断是否使用reverse函数来反转。
3,代码实现
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
*/
class Solution {
public:
vector<vector<int> > Print(TreeNode* pRoot) {
if(!pRoot)
return {};
vector<vector<int>> res;
queue<TreeNode*> q;
q.push(pRoot);
int order = 1;
while(!q.empty()){
int size = q.size();
vector<int> level;
for(int i = 0;i<size;i++){
TreeNode* curr = q.front();
level.push_back(curr->val);
if(curr->left) q.push(curr->left);
if(curr->right) q.push(curr->right);
q.pop();
}
if(!(order%2))
reverse(level.begin(),level.end());
res.push_back(level);
order++;
}
return res;
}
};