【LeetCode】145.二叉树的后序遍历

题目

给你一棵二叉树的根节点 root ，返回其节点值的 后序遍历 。

示例 1：

输入：root = [1,null,2,3]
输出：[3,2,1]

示例 2：

输入：root = []
输出：[]

示例 3：

输入：root = [1,2,3,4,5,6,7]
输出: [4,5,2,6,7,3,1]

提示：

树中节点的数目在范围 [0, 100] 内
-100 <= Node.val <= 100

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

题解

使用深度优先遍历

使用递归方法

/** * 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:
    void fun(TreeNode* node,vector<int> &result)
    {
    
        if(node->left)
            fun(node->left,result);
        if(node->right)
            fun(node->right,result);
        result.emplace_back(node->val);
    }
    vector<int> postorderTraversal(TreeNode* root) {
    
        vector<int> result;
        if(root)
            fun(root,result);
        return result;
    }
};

使用迭代方法

/** * 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) {
    
        vector<int> result;
        stack<TreeNode*> mystack;
        TreeNode* node = root;
        TreeNode* pre = nullptr;

        while(!mystack.empty() || node!=nullptr)
        {
    
            while(node!=nullptr)
            {
    
                mystack.emplace(node);
                node = node->left;
            }

            node = mystack.top();
            mystack.pop();

            if(node->right && node->right!=pre)
            {
    
                mystack.emplace(node);
                node = node->right;
            }
            else
            {
    
                result.emplace_back(node->val);
                pre = node;
                node = nullptr;
            }
        }
        return result;
    }
};