领扣:二叉树后序遍历(非递归)

描述

给出一棵二叉树,返回其节点值的后序遍历。

样例
解答:
了解下后序遍历,在二叉树中,先左后右再根,即首先遍历左子树,然后遍历右子树,最后访问根结点。
那么非递归的思想下,我们可以考虑采用深度优先遍历访问节点的算法来输出后序遍历。由此我们需要借助栈。
同时,我们需要考虑到如何确认节点是否被访问过,因此我们借助set容器保存已经被访问过的节点。容器内的节点表示该节点被访问过。

代码:

/**
 * Definition of TreeNode:
 * class TreeNode {
 * public:
 *     int val;
 *     TreeNode *left, *right;
 *     TreeNode(int val) {
 *         this->val = val;
 *         this->left = this->right = NULL;
 *     }
 * }
 */

class Solution {
public:
    /**
     * @param root: A Tree
     * @return: Postorder in ArrayList which contains node values.
     */
    vector<int> postorderTraversal(TreeNode * root) {
        // write your code here
        std::stack<TreeNode*> tmp;
        std::set<TreeNode*> setUsed;
        vector<int> vec;
        if(root == NULL)
        {
            return vec;
        }
        tmp.push(root);
        
        while(tmp.size())
        {
            TreeNode*& top = tmp.top();
            // 当左孩子未访问过,则将左孩子压入栈
            if(top->left && !setUsed.count(top->left))
            {
                tmp.push(top->left);
                continue;
            }
            // 当右孩子未访问过,则将右孩子压入栈
            if(top->right && !setUsed.count(top->right))
            {
                tmp.push(top->right);
                continue;
            }
			// 左右孩子为空,为当前树的根,压入到后续遍历容器中
            vec.push_back(top->val);
            // 记录已经被访问过
            setUsed.insert(top);
            // 出栈
            tmp.pop();
        }
        
        return vec;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值