【59】按之字形顺序打印二叉树
- 时间限制:1秒
- 空间限制:32768K
题目描述
请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
牛客网题目链接:点击这里
VS2010代码:
// Source: http://www.nowcoder.com/practice/91b69814117f4e8097390d107d2efbe0?tpId=13&tqId=11212&rp=3&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking
// Author: Yang Qiang
// Date : 2016-8-13
#include<iostream>
#include<vector>
using namespace std;
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) {
vector<TreeNode*> a1;
vector<TreeNode*> a2;
vector<int> temp;
vector<vector<int> > result;
if(!pRoot) return result;
bool flag=1; //flag=1,表示先左子树。flag=0,表示先右子树
a1.push_back(pRoot);
temp.push_back(pRoot->val);
result.push_back(temp);
temp.clear();
flag=0;
while(!a1.empty())
{
if(flag==0 && !a1.empty())
{
for(int i=a1.size()-1; i>=0; i--)
{
if(a1[i]->right)
{
a2.push_back(a1[i]->right);
temp.push_back(a1[i]->right->val);
}
if(a1[i]->left)
{
a2.push_back(a1[i]->left);
temp.push_back(a1[i]->left->val);
}
}
a1.clear();
flag=1;
if(!temp.empty())
result.push_back(temp);
temp.clear();
}
if(flag==1 && !a2.empty())
{
for(int i=a2.size()-1; i>=0; i--)
{
if(a2[i]->left)
{
a1.push_back(a2[i]->left);
temp.push_back(a2[i]->left->val);
}
if(a2[i]->right)
{
a1.push_back(a2[i]->right);
temp.push_back(a2[i]->right->val);
}
}
a2.clear();
flag=0;
if(!temp.empty())
result.push_back(temp);
temp.clear();
}
}
return result;
}
};