Leetcode 145二叉树的后序遍历（迭代算法）

给定一个二叉树，返回它的 后序 遍历。

示例:

输入: [1,null,2,3]  
   1
    \
     2
    /
   3 

输出: [3,2,1]

在前中序遍历的基础上增加了一个记录前置结点pre（关于前中序可以看我之前的随笔）。

https://blog.csdn.net/qq_44082048/article/details/120857194

vector<int> ans;
stack<TreeNode*> s;
TreeNode* cur=root;
TreeNode* pre=nullptr;//一个前置结点作为记录

之后我们需要先行将root结点入栈，然后进行循环。

        s.push(root);
        while(!s.empty()){
            cur=s.top();
            if((cur->left==nullptr&&cur->right==nullptr)||(pre!=nullptr&&(pre==cur->left||pre==cur->right))){
                ans.push_back(cur->val);
                s.pop();
                pre=cur;
            }
            else{
                if(cur->right!=nullptr){
                    s.push(cur->right);
                }
                if(cur->left!=nullptr){
                    s.push(cur->left);
                }
            }
        }

cur表示当前结点，pre表示前一个节点。

分析一下判断条件 cur->left==nullptr&&cur->right==nullptr很好判断，当cur不存在左右结点时我们就可以把当前结点加入答案数组，同时这个点就已经没有利用价值了可以把它出栈，它也会变成pre结点。pre==cur->left||pre==cur->right这个判断可能有点难度？其实可以吧这个和 cur->left==nullptr&&cur->right==nullptr条件来等价，已知我们是从栈顶元素开始处理，一个存在左右子结点的结点，它的左右子结点肯定在他的上面，被优先处理。当前一个处理的结点是当前的结点的时候，说明刚刚pop出去的是他的最后一个子结点，所以这个点此时他的左右子结点就能视为空。

 

在就是pre!=nullptr&&(pre==cur->left||pre==cur->right),来逐个分析，首先pre!=nullptr表示的是判断不是最开始的时候

最后注意在压栈的时候优先压右结点再压左结点，因为栈采取的是LIFO,即后进先出，后压入的左结点会被先处理。

最后附上完整代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        if(root==nullptr){
            vector<int> temp={};
            return temp;
        }
        vector<int> ans;
        stack<TreeNode*> s;
        TreeNode* cur=root;
        TreeNode* pre=nullptr;//一个前置结点作为记录
        s.push(root);
        while(!s.empty()){
            cur=s.top();
            if((cur->left==nullptr&&cur->right==nullptr)||(pre!=nullptr&&(pre==cur->left||pre==cur->right))){
                ans.push_back(cur->val);
                s.pop();
                pre=cur;
            }
            else{
                if(cur->right!=nullptr){
                    s.push(cur->right);
                }
                if(cur->left!=nullptr){
                    s.push(cur->left);
                }
            }
        }
        return ans;
    }
};