LeetCode145. 二叉树的后序遍历

最新推荐文章于 2022-04-15 16:29:48 发布

原创最新推荐文章于 2022-04-15 16:29:48 发布 · 623 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode #二叉树后序遍历

C++ 同时被 2 个专栏收录

262 篇文章

订阅专栏

二叉树

27 篇文章

订阅专栏

本文介绍二叉树后序遍历的两种实现方法：递归与非递归。递归方法简洁直观，而非递归方法使用栈来辅助实现，确保节点在其左右子树被访问后再访问自身。

1.递归实现：

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

2.非递归实现：

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

说明：只有一个节点的左右孩子都为空或者左右孩子都被访问过了，才访问当前节点，否则的话，先把右孩子入栈，再把左孩子入栈，因为栈是先进后出的，而后序遍历是先访问左孩子，再访问右孩子，再访问父节点。