请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,
第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
思路:层次遍历采用队列模拟
开始 根节点入队列
开始一个循环,每次出队列一个元素,访问该值
若该节点 左右子节点不为NULL 放入队列尾部
循环直到 队列为空
——如何交替一行一行输出 一开始队列的长度为size表示上一层节点数目 每次控制访问size个,
另外增加一个flag控制左右顺序
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
核心函数
vector<vector<int> > Print(TreeNode* pRoot) {
vector<vector<int>> res;
if(pRoot==NULL)
return res;
vector<TreeNode*> queue;//模拟队列
//首先根节点入队列
queue.push_back(pRoot);
bool flag=false;//从左向右
while(queue.empty()!=1)
{
int size=queue.size();//该层节点数目
vector<int> temp;
int i=0;
while(i<size)
{
TreeNode* p=queue[0];//取队列头
temp.push_back(p->val);
if(p->left!=NULL)
queue.push_back(p->left);
if(p->right!=NULL)
queue.push_back(p->right);
queue.erase(queue.begin());//出队列
i++;
}
if(flag)
{
vector<int> one_res;
for(int i=temp.size()-1;i>=0;i--)
{
one_res.push_back(temp[i]);
}
res.push_back(one_res);
flag=false;//下一次从右向左
}
else
{
res.push_back(temp);
flag=true;
}
}
return res;
}