题目描述
请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
考察点:二叉树和栈的理解;新概念的快速理解。
思路:奇数层顺序输出,偶数层逆序输出。想到了两个栈来存放数据。书上写的有点不太明白,自己写了一个两个栈交替的程序。
奇数层从右节点开始入栈2,偶数层从左孩子开始入栈1。
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<vector<int>> result;
if (pRoot == NULL)
return result;
stack<TreeNode*> stack1;
stack<TreeNode*> stack2;//初始化两个栈
stack1.push(pRoot);//根节点入第一个栈
int flag = 1;//初始化为奇数层
while (!stack1.empty() || !stack2.empty()) //不为空就是有数据,一直循环输出
{
//两个临时vector,存放一行。
vector<int> re1;//奇数行
vector<int> re2;//偶数行
if (flag == 1)
{
while (!stack1.empty())
{
TreeNode* pNode = stack1.top();
stack1.pop();
re1.push_back(pNode->val);//元素出栈入数组
if (pNode->left != NULL) //孩子节点入栈2,先左后右
stack2.push(pNode->left);
if (pNode->right != NULL)
stack2.push(pNode->right);
}
result.push_back(re1);//结果入二维数组
flag = 0; //修改层的标志为偶
}
if (flag == 0)//操作同栈1
{
while (!stack2.empty())
{
TreeNode* pNode = stack2.top();
stack2.pop();
re2.push_back(pNode->val);
if (pNode->right != NULL)//孩子节点入栈1,先右后左
stack1.push(pNode->right);
if (pNode->left != NULL)
stack1.push(pNode->left);
}
if(!re2.empty())//这里需要判断是不到了最后一层。否则会出现讲空元素加入二维数组
result.push_back(re2);
flag = 1;//修改层的标志为奇
}
}
return result;
}
};