LeetCode 145. 二叉树的后序遍历（后序遍历&总结）

Michael阿明

发布于 2021-02-20 14:20:48

29600

代码可运行

文章被收录于专栏：Michael阿明学习之路Michael阿明学习之路

运行总次数：0

代码可运行

1. 题目信息

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

示例:

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

输出: [3,2,1]

进阶: 递归算法很简单，你可以通过迭代算法完成吗？

来源：力扣（LeetCode）链接：https://leetcode-cn.com/problems/binary-tree-postorder-traversal 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解法

2.1 递归

class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> ans;
        preorder(root, ans);
        return ans;
    }
    void preorder(TreeNode* root, vector<int> &ans)
    {
        if(root == NULL)
            return;
        preorder(root->left, ans);
        preorder(root->right, ans);
        ans.push_back(root->val);
    }
};

2.2 循环，必须掌握

左右根

a. 单栈

先按照"根-右-左"的顺序遍历二叉树(和先序遍历有些像)，
然后将遍历的结果反转过来就是“左-右-根”，也就是后序遍历了

class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> ans;
        stack<TreeNode*> stk;
        while(root || !stk.empty())
        {
            while(root)
            {
                stk.push(root);
                ans.push_back(root->val);
                root = root->right;
            }
            root = stk.top()->left;
            stk.pop();
        }
        //反转遍历结果
        reverse(ans.begin(),ans.end());
        return ans;
    }
};

以下解法会破坏二叉树

class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> ans;
        if(root==NULL)
            return ans;
        TreeNode *cur;
        stack<TreeNode*> stk;
        stk.push(root);
        while(!stk.empty())
        {
            cur = stk.top();
            if(cur->left)
            {
                stk.push(cur->left);
                cur->left = NULL;
            }
            else if(cur->right)
            {
                stk.push(cur->right);
                cur->right = NULL;
            }
            else
            {
                ans.push_back(cur->val);
                stk.pop();
            }
        }
        return ans;
    }
};

b. 双栈解法

stk1，模仿前序遍历的实现“反后序遍历”
stk2，保存stk1的pop元素

class Solution {
public:
    vector<int> postorderTraversal(TreeNode *root) {
        vector<int> ans;
        if(root == NULL) return ans;
        stack<TreeNode*> stk1;
        stack<TreeNode*> stk2;
        stk1.push(root);
        TreeNode *cur;
        while(!stk1.empty())
        {
            cur = stk1.top();
            stk1.pop();
            stk2.push(cur);
            if(cur->left)
                stk1.push(cur->left);
            if(cur->right)
                stk1.push(cur->right);
        }
        while(!stk2.empty())
        {
            cur = stk2.top();
            stk2.pop();
            ans.push_back(cur->val);
        }
        return ans;
    }
};