二叉树后序遍历非递归

Posted 2021-06-15 _Camille

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语句中的入栈次序即可。