LeetCode 145-Binary Tree Postorder Traversal（二叉树，后序遍历的迭代写法）

LeetCode 145-Binary Tree Postorder Traversal

题干：

二叉树的后序遍历，用迭代写。

解：

方法1：

先访问右子树的先序遍历（左节点先入栈）得到的ans数组（中右左）反转就是二叉树的后序遍历（左右中）

vector<int> postorderTraversal(TreeNode* root) {
        if(!root) return {};
        vector<int> ans;
        stack<TreeNode*> st;
        st.push(root);
        while(!st.empty()){
            TreeNode* t = st.top();
            st.pop();
            ans.push_back(t->val);
            if(t->left) st.push(t->left);
            if(t->right) st.push(t->right);
        }
        reverse(ans.begin(), ans.end());
        return ans;
    }

方法2：

统一写法。

将访问的节点按递归顺序（后序遍历是左右中，所以入栈顺序是中右左）放入栈中，把要处理的节点也放入栈中但是做好标记。

如何标记？就是要处理的节点放入栈后，紧接着放入一个空指针作为标记。

vector<int> postorderTraversal(TreeNode* root) {
        if(!root) return {};
        stack<TreeNode*> st;
        vector<int> ans;
        st.push(root);
        while(!st.empty()){
            TreeNode* tmp = st.top();
            st.pop();
            if(tmp){
                st.push(tmp); // 中
                st.push(nullptr);
                if(tmp->right) st.push(tmp->right); // 右
                if(tmp->left) st.push(tmp->left); // 左
            }
            else{
                tmp = st.top();
                st.pop();
                ans.push_back(tmp->val);
            }
        }
        return ans;
    }

参考