思路:
对于先序,直接用非递归或者递归的深度搜索方法遍历整个树就可得到;而对于中序及后序,可以事先定义两个栈,用于存储先序遍历的过程中的结点,然后按照各自的次序在该退栈的时候退栈,即可得到中序和后序的遍历顺序。
对于中序:在遍历到左子树的最后一个结点时,将此结点出栈并记录,然后再从此结点的右子节点开始循环遍历即可。
对于后序:退栈的时机与中序相似,在遍历到某个结点的右子节点时,如果这个右子节点为空,则进行退栈操作,需要将此前中序栈退栈过而后序栈没有退栈过的结点全部退栈,直到退到中序栈的栈顶结点为止。
如:
在遍历到结点3,发现左子节点为空,中序栈将3退栈,此时中序栈只剩下结点1,继而发现3的右子节点也为空,则开始对后序栈退栈,此时后序栈内容为1、2、3,退回到栈顶元素等于中序栈的栈顶元素,也就是结点1,则退栈后后序栈也只剩下了结点1,再循环这一步骤,1的右子节点为空,继续退栈直到栈为空,后序遍历顺序为3、2、1.。
代码:
vector<vector<int> > threeOrders(TreeNode* root) {
vector<vector<int> > out_v;
if(!root)
return out_v;
vector<int> first_v,mid_v,last_v;
stack<TreeNode*> mid_order,last_order;
TreeNode* p=root;
TreeNode* p1;
while(!mid_order.empty()||p)
{
while(p)//持续遍历直到左子节点为空
{
first_v.push_back(p->val);
mid_order.push(p);
last_order.push(p);
p=p->left;
}
p=mid_order.top();//开始对中序栈退栈
mid_order.pop();
mid_v.push_back(p->val);
if(!p->right)//右子节点为空则对后序栈退栈
{
p1=mid_order.empty()?NULL:mid_order.top();
while(!last_order.empty()&&last_order.top()!=p1)
{
last_v.push_back(last_order.top()->val);
last_order.pop();
}
}
p=p->right;
}
out_v.push_back(first_v);
out_v.push_back(mid_v);
out_v.push_back(last_v);
return out_v;
}
一次对树的遍历,总体复杂度为O(n)。