二叉树后序遍历非递归

在这里插入图片描述
https://leetcode-cn.com/problems/binary-tree-postorder-traversal/

递归

首先来看递归做法:

class Solution {
public:
    vector<int> ret;
    vector<int> postorderTraversal(TreeNode* root) 
    {
        
        if(root != nullptr)
        {
            postorderTraversal(root->left);
            postorderTraversal(root->right);
            ret.push_back(root->val);
        }
        return ret;
    }
};

二叉树的各种遍历都可以用递归实现,但其效率不是太高。
在这里插入图片描述

迭代

简单说一下迭代思想,先进后出的“栈”,用栈来压入和弹出节点,将树中所有的节点和其左右节点看成一个单位树,如果所有单位树都实现后序遍历,那么整棵树就完成后序遍历。
需要有标记位nullptr,在单位树的根节点入入栈之后,右子树入栈之前入栈,暨出栈的时候该根节点前面若是有一个nullptr,则该单位树已经后序遍历结束。

class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> ret;//存放后序遍历
        stack<TreeNode *> stk;
        if( root == nullptr )//空树
        {
            return ret;
        }
        stk.push(root);

        while( !stk.empty() )
        {
            TreeNode* cur = stk.top(); 
            stk.pop();//获取当前栈顶并出栈。
            
            if( cur!=nullptr )//判断是否是标记位
            {
                stk.push(cur);//不是标记为肯定是实值,入栈,并且在其后再入栈一个标记位
                stk.push(nullptr);
                if(cur->right) stk.push( cur->right );//先入右,再入左,出栈顺序就为先左后右
                if( cur->left ) stk.push( cur->left );
            }
            else//当前为标记位,则此节点(子树根)已经对其操作过
            {//此时标记位已经在前面被pop,那么下一个节点肯定是已经完成遍历的“根”。
                cur = stk.top(); stk.pop();//出栈入vector
                ret.push_back(cur->val);
            }
        }
        return ret;
    }
};

在这里插入图片描述
如果想要实现中序前序遍历,只需要改变if语句中的入栈次序即可。

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值