题目描述
请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
解题思路:
使用两个栈来实现
该题与 剑指offer—把二叉树打印成多行 思路一致,只不过一个是用两个队列来实现,另一个使用两个栈实现。
vector<vector<int> > Print(TreeNode* pRoot){
vector< vector<int> > ret;
if(pRoot == NULL){
return ret;
}
vector<int> v;
stack<TreeNode*> s1,s2;//通过两个栈来处理结点
s1.push(pRoot);
while(!s1.empty() || !s2.empty()){
while(!s1.empty()){
TreeNode* temp = s1.top();
v.push_back(temp->val);
if(temp->left){
s2.push(temp->left);
}
if(temp->right){
s2.push(temp->right);
}
s1.pop();
}
if(!v.empty()){
ret.push_back(v);
v.clear();
}
while(!s2.empty()){
TreeNode* temp = s2.top();
v.push_back(temp->val);
if(temp->right){//此时需要先将右结点入栈
s1.push(temp->right);
}
if(temp->left){
s1.push(temp->left);
}
s2.pop();
}
if(!v.empty()){
ret.push_back(v);
v.clear();
}
}
return ret;
}