[LeetCode] 145. Binary Tree Postorder Traversal

最新推荐文章于 2022-07-19 11:52:33 发布

扑街中的二娃

最新推荐文章于 2022-07-19 11:52:33 发布

阅读量364

点赞数

CC 4.0 BY-SA版权

分类专栏： C++ 日常小题 LeetCode 文章标签： leetcode c++ 日常小题二叉树遍历

本文链接：https://blog.youkuaiyun.com/Elliott_Yoho/article/details/77871598

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

44 篇文章

订阅专栏

日常小题

42 篇文章

订阅专栏

LeetCode

37 篇文章

订阅专栏

本文介绍了一种使用栈实现二叉树后序遍历的非递归算法。通过巧妙地利用两个栈，分别存储节点和节点的状态，实现了无需递归即可完成的后序遍历。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接: https://leetcode.com/problems/binary-tree-postorder-traversal/description/

Description

Given a binary tree, return the postorder traversal of its nodes’ values.

For example:
Given binary tree {1,#,2,3},

return [3,2,1].

Note: Recursive solution is trivial, could you do it iteratively?

解题思路

递归的方式很简单，但是这道题要求用非递归的方式，那么自然想到的就是用栈来解决。由于后序遍历的顺序是 左节点->右节点->根，所以可以先一直沿着左节点往下直到NULL，将这条路径上的所有节点依次放入栈(nStack)中。同时用另一个栈(vStack)，为每一个节点保存状态值，用于弹出栈顶时以辨别是否遍历过其右节点，初始值为false。若已经遍历过其右节点，则同时弹出两个栈的栈顶，并将该节点的值存入返回数组中；若未遍历其右节点，则暂时不弹出，修改 vStack 的栈顶为true，表示已访问过其右节点，并与之前一样，此时的根节点应为该节点的右节点，沿着新的根节点的左节点一直遍历至NULL，重复该过程。

Code

class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> res;
        stack<TreeNode*> nStack;
        stack<bool> vStack;

        while (root != NULL) {
            nStack.push(root);
            vStack.push(false);
            root = root->left;
        }

        while (!nStack.empty()) {
            root = nStack.top();

            if (vStack.top()) {
                res.push_back(root->val);
                nStack.pop();
                vStack.pop();
            } else {
                vStack.top() = true;
                root = root->right;

                while (root != NULL) {
                    nStack.push(root);
                    vStack.push(false);
                    root = root->left;
                }
            }
        }
        return res;
    }
};