实现二叉树先序,中序及后序遍历解题笔记

思路:

对于先序,直接用非递归或者递归的深度搜索方法遍历整个树就可得到;而对于中序及后序,可以事先定义两个栈,用于存储先序遍历的过程中的结点,然后按照各自的次序在该退栈的时候退栈,即可得到中序和后序的遍历顺序。
对于中序:在遍历到左子树的最后一个结点时,将此结点出栈并记录,然后再从此结点的右子节点开始循环遍历即可。
对于后序:退栈的时机与中序相似,在遍历到某个结点的右子节点时,如果这个右子节点为空,则进行退栈操作,需要将此前中序栈退栈过而后序栈没有退栈过的结点全部退栈,直到退到中序栈的栈顶结点为止。
如:
在遍历到结点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)。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值