145. 二叉树的后序遍历

最新推荐文章于 2024-08-24 12:44:53 发布

overlordmax

最新推荐文章于 2024-08-24 12:44:53 发布

阅读量101

点赞数

分类专栏： LeetCode树文章标签：二叉树数据结构 leetcode

本文链接：https://blog.csdn.net/jiangdongxiaobawang/article/details/105242889

版权

LeetCode树专栏收录该内容

51 篇文章 0 订阅

订阅专栏

145. 二叉树的后序遍历

1.题目描述

给定一个二叉树，返回它的后序遍历。
示例:
在这里插入图片描述
进阶: 递归算法很简单，你可以通过迭代算法完成吗？

2.方法1

先序遍历是中左右，后续遍历是左右中，那么我们只需要调整一下先序遍历的代码顺序，就变成中右左的遍历顺序，然后在反转result数组，输出的结果顺序就是左右中了，如下图：
在这里插入图片描述

3.代码

/**
 * 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> res;
        if(root == NULL){
            return res;
        }
        stack<TreeNode*> s;
        s.push(root);
        while(!s.empty()){
            TreeNode* node = s.top();
            s.pop();
            res.push_back(node->val);
            if(node->left){
                s.push(node->left);
            }
            if(node->right){
                s.push(node->right);
            }
        }
        reverse(res.begin(), res.end());
        return res;
    }
};

4.复杂度分析

时间复杂度：O(n)
空间复杂度：O(n)

5.方法2

按照左右中的顺序遍历，一直把root的左节点入栈直到为空
1.此时栈顶元素为树的最左节点，出栈
2.如果最左节点有右子树，则把右子节点入栈，往右子树遍历
3.如果最左节点没有右子树，则把当前节点加入结果数组，并记录当前节点为pre，当节点出栈时，如果节点的右子节点等于pre，则把节点加入结果数组。并把当前节点置空，防止重复遍历。

6.代码

/**
 * 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> res;
        if(root == NULL){
            return res;
        }
        stack<TreeNode*> s;
        TreeNode* pre = NULL;
        while(root != NULL || !s.empty()){
            while(root != NULL){
                s.push(root);
                root = root->left;
            }
            root = s.top();
            s.pop();
            if(root->right == NULL || root->right == pre){
                res.push_back(root->val);
                pre = root;
                root = NULL;
            }
            else{
                s.push(root);
                root = root->right;
            }
        }
        return res;
    }
};