题目描述
请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
解题思路
-
本题目就是二叉树层序遍历的变形,在程序主题中可以定义一个
queue
队列数据结构来方便二叉树的遍历。 -
本题目返回一个动态二维数组即
vector<<vector>int >
类型,思路可以是定义一个一维数组vector<int> tmpVec;
,每次这个二维数组接收这个一维数组即可。 -
由于打印之字形的二叉树,即定义一个
bool even = false;
由于第一行是正序的,所以even
的初始值定义为false
。待遍历第二层结点时,需要将元素逆序,此时even = !even;
即可进行进入元素逆序的部分。 -
每次用
int num = que.size();
类型变量记录当前队列中元素的个数,然后依次将结点值输入给tmpVec
数组,逐一将每个对头结点出队的同时,左右子树进队,以此类推,直到队列中再没有元素为止。
代码实现
/*
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 {
};
bool even = false;
vector<vector<int> > vec; // 多借助其他辅助数据结构
vector<int> tmpVec;
queue<TreeNode*> que;