easy 二叉树的后序遍历递归栈

最新推荐文章于 2022-01-17 22:22:34 发布

踩坑记录

最新推荐文章于 2022-01-17 22:22:34 发布

阅读量85

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/qq_40456702/article/details/119081111

版权

leetcode 专栏收录该内容

90 篇文章 0 订阅

订阅专栏

在这里插入图片描述

递归：

c++


class Solution {
public:
    void postorder(TreeNode* root, vector<int> &res){ // &res给地址
        if(root == nullptr){  //  递归终止的条件为碰到空节点
            return;
        }
        postorder(root->left, res);   // 左
        postorder(root->right, res);  // 右
        res.push_back(root->val);  // 打印根
    }

    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> res;
        postorder(root,res);
        return res;
    }
};

python


class Solution:
    def postorderTraversal(self, root: TreeNode) -> List[int]:
        res=[]
        def postorder(root):
            if not root:
                return res;

            postorder(root.left)
            postorder(root.right)
            res.append(root.val)

        postorder(root)
        return res

在这里插入图片描述

栈：

c++


class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> res;
        if(root == nullptr){
            return res;
        }
        stack<TreeNode*> stk;      // 创建一个栈
        TreeNode* prev = nullptr;  // 记录访问过的右子树
        while(!stk.empty() || root != nullptr){
            // 1.遍历到最左子节点
            while(root){
                stk.push(root);  // 当前遍历到的依次最左子节点入栈
                root = root->left;
            }  // 跳出循环时，当前最左节点下面已经没有节点 root=null
            
            // 2.遍历最左子节点的右子树(右子树存在 && 未访问过)
            root = stk.top();  // 开始回溯，栈顶为最左(底)的节点
            stk.pop();
            // 2.遍历最左子节点的右子树(右子树存在 && 未访问过)
            if(root->right && root->right != prev){
                stk.push(root);
                root = root->right;  // 下个while从该右节点开始找它的最左节点
            }else{   // 没有左右子树，或左右子树已经遍历完 打印
                res.push_back(root->val);
                prev = root;
                root = nullptr;
            }
        }
        return res;
    }
};

python


class Solution:
    def postorderTraversal(self, root: TreeNode) -> List[int]:
        res = []
        if not root:
            return res
        stk = []
        prev = None

        while root or stk:
            ## 找到最左节点(下面已经无左子树 root=null)
            while root:
                stk.append(root)
                root = root.left

            root = stk[-1]   # stak.top()
            if root.right and root.right != prev:
                # 节点有右子树，下个while在右子树继续找最左节点
                root = root.right
            else:  # 节点没有左右子树，或者节点右子树遍历过
                res.append(root.val)
                prev = root   # 遍历过
                root = None   # 遍历过，则下一次不进入while
                stk.pop()
        return res